在 python 中从 csv 文件创建数组

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

但你可以将其保留为 pandas DataFrame。