将数据透视表存储到csv中 - 保留所有数据

时间:2018-05-04 19:13:29

标签: python pandas csv pivot-table

尝试将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的任何建议吗?

0 个答案:

没有答案