怎么做。 我刪除了 pandas df 中列名的不需要的字符？

Question

我有一些以“USI_”開頭的列名的數據集：

我試着用

df.rename(columns = {col: col.strip('USI_') for col in df.columns

但我得到了一些不同的東西。

我預期的 output 應該是“DIVERSITY_QUESTION_RACE”、“DIVERSITY_QUESTION_GENDER”、“UNCOMMON_AFFILIATION”等列

我應該如何修復我的代碼？ 謝謝！

Answer 1

我認為您可能希望將 dataframe 的列名作為列表並通過列表進行字符串操作，然后將這些調整后的列名分配回 dataframe

adj_col_names = []
for name in df.columns:
    adj_col_names.append(name.replace('USI_',''))

df.columns = adj_col_names

Answer 2

嘗試使用替換 function

df.columns = df.columns.str.replace('USI_','')

Answer 3

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