python - 如何从 Pandas DataFrame header 中去除空格？

Question

我正在解析 Excel 文件中的数据，该文件的某些列标题中有多余的空格。

当我使用 df.columns 检查生成的数据框的列时，我看到:

Index(['Year', 'Month ', 'Value'])
                     ^
#                    Note the unwanted trailing space on 'Month '

因此，我做不到:

df["月份"]

因为它会告诉我找不到该列，因为我要求的是“月”，而不是“月”。

那么，我的问题是如何去除列标题中不需要的空白？

Answer 1

您可以将函数赋予 rename 方法。 str.strip() 方法应该做你想做的事:

In [5]: df
Out[5]: 
   Year  Month   Value
0     1       2      3

[1 rows x 3 columns]

In [6]: df.rename(columns=lambda x: x.strip())
Out[6]: 
   Year  Month  Value
0     1      2      3

[1 rows x 3 columns]

注意:这将返回一个 DataFrame 对象，并在屏幕上显示为输出，但实际上并未在您的列上设置更改。要进行更改，请在方法链中使用它或重新分配 df 变量:

df = df.rename(columns=lambda x: x.strip())