替換pandas df列名中的字符串

Question

我在pandas中有一個數據框，列名為“string_string”，我試圖通過刪除“_”和后面的字符串來重命名它們。 例如，我想將“12527_AC9E5”更改為“12527”。 我試圖使用各種替換選項，我可以替換字符串的特定部分（例如，我可以替換所有“_”），但是當我引入通配符時，我沒有達到預期的結果。

以下是我認為可行的一些事情，但事實並非如此。 如果我刪除他們工作的外卡（即，他們替換_）。

df = df.rename(columns=lambda x: x.sub('_.+', ''))

df.columns = df.columns.str.replace('_.+','')

任何幫助贊賞

Answer 1

只需拆分'_'並取第一個元素。 你可以利用字典理解：

df = df.rename(columns={col: col.split('_')[0] for col in df.columns})