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

Question

当试图将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=',')

打印语句的结果如下：

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_表的结果正确存储到csv，有什么建议吗？你知道吗