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'])

Answer 2

您可以添加内容skiprow=0以防止recfromcsv跳过第一行。