如何用panda替换数据帧中的部分标签

Question

我有一个从.csv加载的数据帧，我想删除标签中的一些文本。

现在，我的数据帧将标签保存为output.text.user.12、output.txt.user.1224、，。。。我想从标签中删除零件"output.text.user.">

output.text.user.12  ...  output.text.user.23424
index                        ...                                 
332                 0.06924  ...                     0.0

有人知道我该怎么做吗？我已经看到了如何从字典中替换当前名称，但这是一个太长的数据帧

Answer 1

DataFrame.rename就是您想要的。假设您的数据帧是df

df = df.rename(columns=lambda x: x.replace('output.text.user.', ''))

Answer 2

考虑以下片段：

import pandas as pd
# your dataframe
df = pd.DataFrame()
# loop over columns, split by dot (.) and select last item in resulting list
new_columns = []
for column in df.columns:
new_columns.append(column.split('.')[-1])
# assign new column names to your dataframe by overwriting the old ones
df.columns = new_columns