在Pandas中转换多个值列

一种可能的解决方案,其步骤如下:

• 第一步是设置数据帧的索引 df 到列 YEAR 和 CD_PRD 使用 set_index 方法。

• 那么 stack 方法用于将列转换为行,从而创建层次索引。

• 这个 unstack 然后应用该方法将索引的最内层转换回列中。

• 这个 reset_index 方法用于重置数据帧的索引,将索引转换回列。

• 这个 rename_axis 方法用于删除轴的名称。

• 这个 set_axis 方法用于将数据帧的列重命名为列表 colnames .

• 最后,the pipe 方法用于应用一个lambda函数,该函数使用 assign 创建新列的方法 MONTH 通过删除前缀 VL_ 从中的值 月份 列使用 str.replace 方法。

colnames = ['YEAR', 'MONTH', 'VL_CD76', 'VL_CD92']

(df.set_index(['YEAR', 'CD_PRD']).stack()
 .unstack(1).reset_index().rename_axis(None, axis=1)
 .set_axis(colnames, axis=1)
 .pipe(lambda x: x.assign(MONTH = x['MONTH'].str.replace('VL_', ''))))

输出:

    YEAR MONTH  VL_CD76  VL_CD92
0   2022   JAN    10000    15000
1   2022   FEB     8000     7500
2   2022   MAR   170000    86000
3   2022   APR    58000     1200
4   2023   JAN     8000    12000
5   2023   FEB    12000     4560
6   2023   MAR    15000    10480
7   2023   APR     5000    15600
8   2024   JAN      400    56000
9   2024   FEB     8000    10200
10  2024   MAR    45800   150000
11  2024   APR      120     1500