动态重命名列 Python

Question

我有一个包含以下列（大约 4000）的数据框：

QA1_1, Q8_r1_c3_1, Q19b_5_1 , ... , QA1_32, Q8_r1_c3_32, Q19b_5_32

我创建了两个词典，一个是我想在“_”之前用我的变量替换的名称，另一个是从 _1 到 _32 的相关结尾，例如：

dict_1 = {'QA1' : 'electric', 
          'Q8_r1_c3' : 'solar',
                    ...
           'Q19b_5' : 'urban'}

dict_2 = {'_1' : 'Restaurants',
          '_2' : 'Hotels',
                 ...
          '_32' : 'School'}

我的问题是：如何重命名我的列以与我的变量的一般名称以及相关的后缀一致

期望的最终结果：

electric_Restaurants , solar_Restaurants, urban_Restaurants , ... , electric_School , solar_School, urban_School

Answer 1

使用str.replace：

df.columns = df.columns.str.replace(r'(.*)_([^_]+)$', lambda m: f'{dict_1.get(m.group(1), m.group(1))}_{dict_2.get(m.group(2), m.group(2))}')