pandas 出现 KeyError 问题解决

在使用df['列名'] 时出现了KeyError 问题,但是该列明明是存在的，通过查询资料后，发现是自己列名中包含了空格，所以导致无法访问。在这里记录下如何发现空格，以及解决的方法。

使用一下语句访问时出现KeyError 错误：

import pandas as pd
df = pd.read_csv('1site.csv')
print(df['age']

查资料后发现可能是空格导致，检查是否含有空格：

import pandas as pd
df = pd.read_csv('1site.csv')
cols = df.columns
cols_spaces = [col for col in cols if ' ' in col]
if cols_spaces:
  print("列名中包含空格的列有：", cols_spaces)
else:
  print("所有列名均不包含空格。")

最后发现是哪些列包含空格，但是为了方便，直接将全部列的空格删除

df.rename(columns=lambda x: x.strip().replace(" ", "_"), inplace=True)

此时最初始的数据源并没有发生改变，如果要修改数据源的则将结果保存到新的数据源中

df.to_csv('new_site.csv') # 生成新的文件 new_site.csv