如何用熊貓替換數據框中的部分標簽？

Question

我有一個從 .csv 加載的數據框，我想刪除標簽中的一些文本。

現在，我的數據框將標簽保存為output.text.user.12 ， output.text.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