使用正则表达式为Pandas重命名数据框中的列

Question

我有一个由Pandas制作的数据框,我想删除每个列名末尾的空白区域.我尝试过类似的东西:

raw_data.columns.values = re.sub(' $','',raw_data.columns.values)

但这不起作用,我在这里做错了什么？

Answer 1

我应该使用这个re包:

raw_data = raw_data.rename(columns=lambda x: re.sub(' $','',x))

Answer 2

我建议使用pandas.Series.str.strip

df.columns = df.columns.str.strip()

Answer 3

@Christian 的答案可能适合这个特定问题，但对于有关替换列中名称的更一般问题，我建议创建一个字典理解并将其传递给重命名函数：

df.rename(columns={element: re.sub(r'$ (.+)',r'\1', element, flags = re.MULTILINE) for element in df.columns.tolist()})

就我而言，我想在每列的开头添加一些内容，因此：

df.rename(columns={element: re.sub(r'(.+)',r'x_\1', element) for element in df.columns.tolist()})

您可以使用 inplace=True 参数来实际在数据框中进行更改。