Как заменить часть метки в кадре данных на pandas?

Question

У меня есть фрейм данных, который загружен из .csv, и я хотел бы удалить некоторый текст в метках.

В данный момент мой фрейм данных сохраняет метки как вывод. 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