np.loadtxt 忽略标头，如何保存标头数据？

Question

我使用 savetxt 保存了一个 numpy 数组，并给了该数组一个标题。当我使用 loadtxt 读取文件时，标题将被忽略，只有数据保存在我的新数组中。我如何访问标题，因为它包含我想保存为字符串的重要信息。

编辑：

np.savetxt(file_name, array, delimiter=",", header='x,y,z, data from monte carlo simulation')
data = np.loadtxt('test', dtype=float, delimiter=',')

我想获取“来自蒙特卡洛模拟的数据”并将其保存为字符串。

Answer 1

Mal*_*eau 4

要获取标题，您只需使用.readline()文件上的方法读取文件的第一行即可。在你的情况下它看起来像这样：

f = open(filename)
header = f.readline()
last_col_name = header.split(',')[-1] #returns 'data from monte carlo simulation'

另外，如果您想研究一种更通用的数据存储方式，您可以查看 pandas 库。