python - numpy recfromcsv 和 genfromtxt 跳过数据文件的第一行

Question

numpy 的 recfromcsv 跳过了我数据的第一行。 (对于 genfromtxt 也是一样)

import numpy as np

filename = 'data.csv'
writer = open(filename,mode='w')
writer.write('0,1.1,1.2\n1,2.1,2.2\n2,3.1,3.2')
writer.close()

data = np.recfromcsv(filename)
print data

这是一个错误，还是我如何在不丢失第一行的情况下加载数据？

Answer 1

csv 文件的默认第一行包含字段名称。recfromcsv 函数默认调用 genfromtxt 参数 names=True。这意味着它读取数据的第一行作为标题。

定义: http://docs.scipy.org/doc/numpy/reference/generated/numpy.genfromtxt.html

应该写在数组之前。

import numpy as np

filename = 'data.csv'
writer = open(filename,mode='w')
writer.write('first column,second column,third column\n')
writer.write('0,1.1,1.2\n1,2.1,2.2\n2,3.1,3.2')
writer.close()

data = np.recfromcsv(filename)
print data

或者使用recfromtxt代替recfromcsv。

或者覆盖默认名称为

recfromcsv(filename, names=['a','a','a'])