如何用Python将多列重复列名数据转换为长格式并提取步骤信息？-Python教程-PHP中文网

如何用Python将多列重复列名数据转换为长格式并提取步骤信息？

碧海醫心

发布： 2025-03-01 14:38:12

原创

931人浏览过

如何用python将多列重复列名数据转换为长格式并提取步骤信息？

利用Python Pandas库高效处理多列重复列名数据，将其转换为长格式并提取步骤信息。本文将详细讲解如何实现这一数据转换，并提供具体的代码示例。

问题描述：

许多数据分析任务需要将宽格式数据转换为长格式，以便于后续分析。本文将解决一个实际问题：如何使用Python的Pandas库，将具有重复列名模式（例如，step4_nm_stdedev，step6_nm_stdedev，step8_nm_stdedev）的数据集转换为长格式，并提取步骤信息（step）为单独的一列。假设数据集包含多个步骤的数据，每个步骤对应一组相同的指标（例如，nm_stdedev）。这些指标在数据集中以类似 stepX_metricY 的列名表示，其中X代表步骤号，metricY代表具体的指标。目标是将宽格式数据集转换为长格式，包含步骤号（step）和指标数据列。

解决方案：

立即学习“Python免费学习笔记（深入）”；

Pandas的melt函数结合字符串操作可以高效地解决这个问题。以下代码展示了具体的转换过程：

import pandas as pd

# 示例数据 (请替换为你的实际数据)
data = {'Step4_nm_Stdedev': [1, 2, 3],
        'Step6_nm_Stdedev': [4, 5, 6],
        'Step8_nm_Stdedev': [7, 8, 9],
        'Step4_another_metric': [10, 11, 12],
        'Step6_another_metric': [13, 14, 15],
        'Step8_another_metric': [16, 17, 18]}
df = pd.DataFrame(data)

# 使用melt函数将数据转换为长格式
df_melted = pd.melt(df, var_name='Step_Metric', value_name='Value')

# 使用字符串操作提取Step和Metric信息
df_melted[['Step', 'Metric']] = df_melted['Step_Metric'].str.split('_', expand=True)
df_melted['Step'] = df_melted['Step'].str.replace('Step', '').astype(int)

# 最终结果
print(df_melted)

登录后复制

代码首先使用melt函数将宽格式数据转换为长格式，Step_Metric列包含步骤和指标信息，Value列包含对应的数据值。然后，通过字符串的split方法将Step_Metric列拆分为Step和Metric两列，并使用replace和astype方法清理Step列，去除"Step"前缀并转换为整数类型。最终得到一个包含Step、Metric和Value三列的长格式数据集，实现了数据的有效转换。请根据实际列名和数据进行相应的调整。

以上就是如何用Python将多列重复列名数据转换为长格式并提取步骤信息？的详细内容，更多请关注php中文网其它相关文章！