如何使用pandas从csv读取特定列索引

Question

有没有办法使用Pandas(最好是read_csv)从csv文件中只读取具有特定索引的特定列？我知道read_csv提供了按列名读取特定列的能力，但是数据文件没有标题，所以我不能使用列名。注意，文件太大了，所以我不想读入整个文件，然后再读入子集。谢谢。

Answer 1

这里有一个例子说明了EdChum给出的答案。加载CSV文件有很多额外的选项，请检查API reference。

raw_data = {'first_name': ['Steve', 'Guido', 'John'],
        'last_name': ['Jobs', 'Van Rossum', "von Neumann"]}
df = pd.DataFrame(raw_data)
# Saving data without header
df.to_csv(path_or_buf='test.csv', header=False)
# Telling that there is no header and loading only the first name
df = pd.read_csv(filepath_or_buffer='test.csv', header=None, usecols=[1], names=['first_name'])
df

  first_name
0      Steve
1      Guido
2       John

Answer 2

import pandas as pd
data = pd.read_csv('file.csv', usecols=['column_name'])

usecols参数包含列名称列表。如果想要多个列，则用逗号分隔它们，即['column_name1, 'column_name2', 'column_name3']