我有一个数据框。如果倒数第二个字符串是“pi”，我想交换由“_”分隔的最后两个列字符串

数据框具有如下列：

     abc_rte        abc_rte_log     abc_rte_log_pi1     abc_rte_pi1_log     xyz_pnct_pi2_log

所需的列名：

      abc_rte       abc_rte_log     abc_rte_log_pi1     abc_rte_log_pi1     xyz_pnct_log_pi2    

到目前为止我尝试过的：

        for i in range(0, len(df.columns)):
          if str(df.columns[i].split('_')[-2] == 'pi':
            df.columns[i].split('_')[-2] = str(df.columns[i].split('_')[-1])

回答

df.columns = df.columns.str.replace(r'(pid*)_([^_]+)$', r'2_1')

>>> df.columns

Index(['abc_rte', 'abc_rte_log', 'abc_rte_log_pi1', 'abc_rte_log_pi1',
       'xyz_pnct_log_pi2'],
      dtype='object')