将pandas数据帧保存到csv时如何保留columns.name?

时间:2014-08-06 02:42:57

标签: python csv numpy pandas ipython

初始问题

当我在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

这是一个示例表:https://github.com/SimonBiggs/Electron-Cutout-Factors/blob/master/output/20140807_173714/06app06eng/interpolation-table.csv

2 个答案:

答案 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