python - Pandas 读取错误的列

Question

我有一个 csv 文件，其中包含句子、长度、类别和另外 18 列。我正在尝试过滤掉特定的列。

假设我有 x,y,a,b,c,d,e,f,g,h 作为最后 10 列。我正在尝试过滤掉长度、类别和最后八列。

当我单独对最后 8 列执行此操作时，

col_req = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
data = pd.read_csv('data.csv', names=col_req)

它工作得很好。但当我尝试时，

col_req = ['length','category','a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
data = pd.read_csv('data.csv', names=col_req) 

输出是，

('g', 'h', 'x', 'y', 'a', 'b', 'c', 'd', 'e', 'f')

我不知道我错在哪里。

Answer 1

您需要使用参数 use_cols 来执行此操作

 col_req = ['length','category','a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
 data = pd.read_csv('data.csv', use_cols=col_req)