尝试在列中查找字符串时出错

Question

我正在尝试在数据框列中查找 ID。

如果存在，那么新的 df 将转换为对应的 df，但会引发错误。

哪里可能出错？

df = pd.read_excel('somefile.xlsx')
ids = df['ID'].to_list()

for id in ids:
  if id in ids:
    df_arc = df_arc.set_index('ID')
    df_arc=df_arc.loc[id]
  else:
    print('It is not here')

这是我收到的错误：

KeyError: "None of ['ID'] are in the columns"

Answer 1

df["ID"] 将尝试访问您没有的 ID 列。

如果您引用行索引，那么您可以更改此语句：

ids = df.index.to_list()