сохранить сводную таблицу в CSV - сохранить все данные - PullRequest
Новая
       8

сохранить сводную таблицу в CSV - сохранить все данные

0 голосов
/ 04 мая 2018

При попытке сохранить результат pivot_table в CSV, он не сохраняет все данные (заголовки)

Вот пример кода: 

data = [{'TestPriority': 3, 'AutomationStatus': 'Done'}, 
{'TestPriority': 4, 'AutomationStatus': 'Review'}, {'TestPriority': 2, 
'AutomationStatus': 'Review for Automation'}, {'TestPriority': 2, 
'AutomationStatus': 'Review for Automation'}]

df = pd.DataFrame(data)
df['AutomationStatus'] = np.where(
    df['AutomationStatus'] == 'Done', 'Automated', 'Manual')
res = df.pivot_table(
    index='AutomationStatus', columns=['TestPriority'],
    aggfunc=len, fill_value=0)
print(res)


res.to_csv('sample.csv', sep=',')

результат оператора print выглядит так: 

TestPriority      2  3  4
AutomationStatus         
Automated         0  1  0
Manual            2  0  1

пока сгенерированный csv не показывает заголовок TestPriority: 

AutomationStatus,2,3,4
Automated,0,1,0
Manual,2,0,1

Любые предложения о том, как правильно сохранить результат pivot_table в CSV без потери данных?

...