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