python - 为什么我使用 pandas 从中读取 csv 文件的对象是 TextFileReader 对象

Question

我使用 pandas 读取了一个 csv 文件:

data_raw = pd.read_csv(filename, chunksize=chunksize)
print(data_raw['id'])

然后，它报告 TypeError:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'TextFileReader' object has no attribute '__getitem__'

我该怎么做才能解决这个问题？以及如何将 data_raw 更改为 dataFrame 对象？我用的是python2.7和pandas v0.19.1

Answer 1

解决此问题的一种方法是在 pd.read_csv() 函数中设置 nrows 参数，这样您就可以选择要加载到数据框中的数据子集。当然，缺点是您将无法查看和使用完整的数据集。代码示例:

data = pd.read_csv(filename, nrows=100000)