使用字符串标题将numpy数组保存到csv

Question

我有一个包含浮点数的大型numpy数组,我将其保存为带有np.savetxt的csv文件("myFile.csv",myArray,delimiter =",")

现在,由于数组有很多列而且很难记住它是什么,我想在导出之前向数组中添加一个字符串头.由于numpy不接受float数组中的字符串,有没有一个技巧来完成这个？

[解决方案]感谢Cyborg的建议,我设法安装了Pandas.

import pandas as pd
df = pd.DataFrame(A) # A is a numpy 2d array
df.to_excel("A.xls", header=C,index=False) # C is a list of string corresponding to the title of each column of A

Answer 1

该header参数(文档):

numpy.savetxt(fname, X, delimiter=' ', header='')

但是如果你实际上正在处理一张桌子,你可能更喜欢熊猫.