python - np.loadtxt 忽略 header ，如何保存 header 数据？

Question

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

编辑:

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

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

Answer 1

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

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

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