KeyError：“ [[”，“]]]都不在“ [列]”中

Question

我想在我的数据框中切两列。

这是我这样做的代码：

import pandas as pd
df = pd.read_csv('source.txt',header=0)
cidf=df.loc[:,['vocab','sumCI']]
print(cidf)

这是数据示例：

ID  vocab   sumCI   sumnextCI   new_diff
450      statu    3.0        0.0       3.0
391     provid    4.0        1.0       3.0
382  prescript    3.0        0.0       3.0
300   lymphoma    2.0        0.0       2.0
405      renew    2.0        0.0       2.0

**首先我收到此错误：**

KeyError: “None of [['', '']] are in the [columns]”'

我尝试过的

• 我在读取文件时尝试header用index 0

• 我尝试使用以下代码重命名列：

  df.rename(columns=df.iloc[0],inplace=True)
  

• 我也试过这个：

  df.columns = df.iloc[1]
  df=df.reindex(df.index.drop(0))
  

• 在此链接中也尝试过评论

以上都不是解决问题的方法。

Answer 1

通过您发布的打印，似乎您有空白作为分隔符。pd.read_csv将使用,默认分隔符读取，因此您必须明确声明：

pd.read_csv('source.txt',header=0, delim_whitespace=True)