当我在ipython
中运行以下内容时import numpy as np
import pandas as pd
df = pd.DataFrame(np.round(9*np.random.rand(4,4), decimals=1))
df.index.name = 'x'
df.columns.name = 'y'
df.to_csv('output.csv')
df
它输出以下结果:
y 0 1 2 3
x
0 7.6 7.4 0.3 7.5
1 5.6 0.0 1.5 5.9
2 7.1 2.1 0.0 0.9
3 3.7 6.6 3.3 8.4
但是,当我打开output.csv
时,“y”被删除:
x 0 1 2 3
0 7.6 7.4 0.3 7.5
1 5.6 0 1.5 5.9
2 7.1 2.1 0 0.9
3 3.7 6.6 3.3 8.4
如何在将数据框输出到csv时保留df.columns.name
?
目前的粗略解决方法是我做以下事情:
df.to_csv('output.csv', index_label = 'x|y')
这导致output.csv
阅读:
x|y 0 1 2 3
0 7.6 7.4 0.3 7.5
1 5.6 0 1.5 5.9
2 7.1 2.1 0 0.9
3 3.7 6.6 3.3 8.4
更好的东西会很棒!感谢您的帮助(提前)。
这就是我正在处理的问题:https://github.com/SimonBiggs/Electron-Cutout-Factors
答案 0 :(得分:4)
您可以传递列表来命名列,然后在写入csv时指定索引名称:
df.columns = ['column_name1', 'column_name2', 'column_name3']
df.to_csv('/path/to/file.csv', index_label='Index_name')
答案 1 :(得分:2)
这个怎么样?它略有不同但希望可用,因为它符合CSV范例:
>>> df.columns = ['y{}'.format(name) for name in df.columns]
>>> df.to_csv('output.csv')
>>> print open('output.csv').read()
x,y0,y1,y2,y3
0,3.5,1.5,1.6,0.3
1,7.0,4.7,6.5,5.2
2,6.6,7.6,3.2,5.5
3,4.0,2.8,7.1,7.8