如何从Pandas DataFrame标头中删除空格？

Question

我正在从Excel文件中解析数据，该文件在某些​​列标题中具有额外的空白。

当我使用来检查结果数据框的列时df.columns，我看到：

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

因此，我无法执行以下操作：

df["Month"]

因为它会告诉我找不到列，因为我要求输入“ Month”而不是“ Month”。

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

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对象，并在屏幕上显示为输出，但是更改实际上并未设置在您的列上。要进行更改，请使用：

1. 使用inplace=True参数[docs]

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

2. 将其分配回您的df变量：

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