[英]how to write user-defined column names to a csv file using pandas?
將透視表寫入CSV文件時,可以寫自己的列名嗎?
import pandas as pd
c1 = ['a','a','a','a','b','b','b','b']
c2 = ['x','x','y','y','x','x','y','y']
c3 = [1, 2, 1, 2, 1, 2, 1, 2]
val = [85, 47, 29, 93, 15, 21, 65, 16]
df = pd.DataFrame({'c1':c1, 'c2':c2, 'c3':c3, 'val':val})
ptable = pd.pivot_table(data=df, cols=['c2','c3'], rows='c1')
我嘗試使用header
參數:
ptable.to_csv('test.csv', header=['n1','n2','n3','n4'])
但列名沒有改變...
解決方法:在調用to_csv
之前更改ptable
的列:
ptable = pd.pivot_table(data=df, cols=['c2','c3'], rows='c1')
ptable.columns = ['n1','n2','n3','n4']
ptable.to_csv('/tmp/test.csv')
只是在寫之前重命名:
ptable.columns=['n1','n2','n3','n4']
ptable.to_csv(r'c:\data\test.csv')
實際上,它應該通過傳遞header
參數的列表來工作,但不確定為什么,這可能是一個錯誤
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.