在python中从cvs文件创建数组

Question

所以我有一个数据文件，我必须从中提取特定的数据。使用

x=15 #need a way for code to assess how many lines to skip from given data
maxcol=2000 #need a way to find final row in data
data=numpy.genfromtxt('data.dat.csv',skip_header=x,delimiter=',')
column_one=data[0;max,0]
column_two=data[0:max,1]

这为特定情况提供了一个数组，其中所需数据上方有（x=）15行元数据，数据行数为（maxcol=）2000。我该如何更改代码以满足x和maxcol的任何值？你知道吗

Answer 1

使用pandas。它的read_csv函数做了您想要的一切（我不包括它的等价物delimiter，sep=','，因为逗号分隔是默认值）：

import pandas as pd
data = pd.read_csv('data.dat.csv', skiprows=x, nrows=maxcol)

如果您确实希望将其作为numpy数组，可以执行以下操作：

data = data.values

但是你可以把它当作熊猫。你知道吗