np.loadtxt忽略标题,如何保存标题数据?

时间:2018-08-31 10:23:04

标签: python numpy

我已经使用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=',')

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

1 个答案:

答案 0 :(得分:1)

要获取标题,您可以使用文件上的.readline()方法简单地读取文件的第一行。就您而言,它看起来像这样:

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

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