pandas如何对DataFrame执行reshaping变换得到目标输出结构

你已经通过melt得到了长格式的中间结果，接下来只需要对分组内的记录生成序号，再做宽表转换即可得到目标输出，完整实现代码如下：

import pandas as pd

# 你的原始melt操作
melted = pd.melt(df, id_vars=['col_name'], var_name = 'NAME', value_name = 'VALUE')
# 按col_name和NAME分组，给组内每行生成序号，对应后续的col1/col2/col3
melted['idx'] = melted.groupby(['col_name', 'NAME']).cumcount() + 1
# 透视转换为宽表
result = melted.pivot(index=['col_name', 'NAME'], columns='idx', values='VALUE').reset_index()
# 重命名列名符合要求
result.columns = ['col_name', 'NAME'] + [f'col{i}' for i in range(1, len(result.columns)-1)]

运行上述代码后得到的result就是你需要的目标格式。

内容的提问来源于stack exchange，提问作者sparkstars