将行数据转换为列:Pandas pivot 方法详解

花韻仙語
发布: 2025-11-20 13:04:18
原创
447人浏览过

将行数据转换为列:Pandas pivot 方法详解

本教程详细介绍了如何使用 pandas 的 `pivot` 方法将数据框中按行存储的页面级信息转换为按列展示的报告级汇总数据。通过指定索引、列和值参数,结合 `add_prefix`、`reset_index` 和 `rename_axis` 等辅助操作,实现数据重塑,将不同页码的值转换为独立的列,从而高效地聚合和展示复杂的数据结构,同时处理不同报告页数不一致的情况。

在数据分析和处理中,我们经常会遇到需要将“长格式”数据(即某些属性的值作为行来存储)转换为“宽格式”数据(即这些属性的值作为新的列来存储)的场景。例如,当一份报告包含多个页面的数据,并且每个页面的信息都作为独立的一行存储时,我们可能希望将同一报告(由公司和年份标识)的所有页面数据聚合到一行中,并为每个页面创建一个独立的列。Pandas 库提供了强大的 pivot 函数来高效地完成这类数据重塑任务。

1. 问题背景与数据示例

假设我们有一份关于公司年度报告的数据,其中包含了公司(FIRM)、年份(YEAR)、报告页码(Report Page)以及对应的数值(Value1)。初始数据以长格式存储,每个报告的每个页面都占据一行。

原始数据结构:

import pandas as pd

data = {
    'FIRM': ['A', 'A', 'B', 'B'],
    'YEAR': [2012, 2012, 2013, 2013],
    'Report Page': [1, 2, 1, 2],
    'Value1': [10, 15, 20, 25]
}

df = pd.DataFrame(data)
print("原始数据框:")
print(df)
登录后复制

输出结果:

原始数据框:
  FIRM  YEAR  Report Page  Value1
0    A  2012            1       10
1    A  2012            2       15
2    B  2013            1       20
3    B  2013            2       25
登录后复制

我们的目标是将每个报告(由 FIRM 和 YEAR 唯一确定)的页面级信息(Value1)从行转换为列。具体来说,我们希望 Report Page 1 的 Value1 成为 Value1_Page1 列,Report Page 2 的 Value1 成为 Value1_Page2 列,以此类推。

期望的输出结构:

  FIRM  YEAR  Value1_Page1  Value1_Page2
0    A  2012            10            15
1    B  2013            20            25
登录后复制

2. 使用 pandas.pivot 进行数据重塑

pandas.pivot 函数是实现这种行转列操作的核心工具。它接收三个关键参数:

  • index: 用于构建新数据框索引的列或列的列表。
  • columns: 用于构建新数据框列的列。
  • values: 用于填充新数据框值的列或列的列表。

针对我们的需求,我们可以这样使用 pivot:

  • index=['FIRM', 'YEAR']: 这两列将作为新数据框的复合索引,唯一标识每个报告。
  • columns='Report Page': Report Page 列的值将成为新数据框的列名。
  • values='Value1': Value1 列的值将填充到新数据框中。

基本 pivot 操作:

# 步骤1: 使用 pivot 进行重塑
df_pivoted = df.pivot(index=['FIRM', 'YEAR'], columns='Report Page', values='Value1')
print("\n步骤1: 初始 pivot 结果:")
print(df_pivoted)
登录后复制

输出结果:

步骤1: 初始 pivot 结果:
Report Page    1   2
FIRM YEAR           
A    2012     10  15
B    2013     20  25
登录后复制

此时,我们已经将 Report Page 的值转换为列,并且 Value1 的值填充到了相应的位置。然而,列名仍是页码本身(1, 2),且 FIRM 和 YEAR 仍是索引。

3. 后续处理与优化

为了达到最终的期望格式,我们还需要进行一系列的辅助操作:

3.1 添加列名前缀 (add_prefix)

为了使列名更具描述性,我们可以使用 add_prefix() 方法为新生成的列名添加一个前缀。

电子手机配件网站源码1.0
电子手机配件网站源码1.0

电子手机配件网站源码是一个响应式的织梦网站模板,软件兼容主流浏览器,且可以在PC端和手机端中进行浏览。模板包含安装说明,并包含测试数据。本模板基于DEDECms 5.7 UTF-8设计,需要GBK版本的请自己转换。模板安装方法:1、下载最新的织梦dedecms5.7 UTF-8版本。2、解压下载的织梦安装包,得到docs和uploads两个文件夹,请将uploads里面的所有文件和文件夹上传到你的

电子手机配件网站源码1.0 0
查看详情 电子手机配件网站源码1.0
# 步骤2: 添加列名前缀
df_prefixed = df_pivoted.add_prefix('Value1_Page')
print("\n步骤2: 添加列名前缀后:")
print(df_prefixed)
登录后复制

输出结果:

步骤2: 添加列名前缀后:
             Value1_Page1  Value1_Page2
FIRM YEAR                              
A    2012              10            15
B    2013              20            25
登录后复制

3.2 重置索引 (reset_index)

pivot 操作会将 index 参数指定的列设置为新的索引。为了将 FIRM 和 YEAR 重新作为普通的数据列,我们需要使用 reset_index() 方法。

# 步骤3: 重置索引
df_reset = df_prefixed.reset_index()
print("\n步骤3: 重置索引后:")
print(df_reset)
登录后复制

输出结果:

步骤3: 重置索引后:
FIRM  YEAR  Value1_Page1  Value1_Page2
0    A  2012            10            15
1    B  2013            20            25
登录后复制

3.3 清理列轴名称 (rename_axis)

在 reset_index() 之后,原来的 columns 参数 (Report Page) 会成为列轴的名称,显示在数据框的左上角。为了使数据框更整洁,我们可以使用 rename_axis(None, axis=1) 将其移除。

# 步骤4: 清理列轴名称
df_final = df_reset.rename_axis(None, axis=1)
print("\n步骤4: 清理列轴名称后 (最终结果):")
print(df_final)
登录后复制

输出结果:

步骤4: 清理列轴名称后 (最终结果):
  FIRM  YEAR  Value1_Page1  Value1_Page2
0    A  2012            10            15
1    B  2013            20            25
登录后复制

4. 完整解决方案代码

将上述所有步骤合并,可以得到简洁高效的解决方案:

import pandas as pd

data = {
    'FIRM': ['A', 'A', 'B', 'B'],
    'YEAR': [2012, 2012, 2013, 2013],
    'Report Page': [1, 2, 1, 2],
    'Value1': [10, 15, 20, 25]
}

df = pd.DataFrame(data)

# 一步完成数据重塑和清理
df_result = df.pivot(index=['FIRM', 'YEAR'], columns='Report Page', values='Value1') \
              .add_prefix('Value1_Page') \
              .reset_index() \
              .rename_axis(None, axis=1)

print("最终重塑的数据框:")
print(df_result)
登录后复制

5. 处理不同报告页数不一致的情况

教程中提到“Reports do not always have the same number of pages.”(报告的页数并不总是相同的)。pandas.pivot 能够优雅地处理这种情况。如果某个报告缺少某个页码的数据,pivot 会自动在该位置填充 NaN(Not a Number)。

示例:一个报告只有一页

import pandas as pd

data_uneven = {
    'FIRM': ['A', 'A', 'B'],
    'YEAR': [2012, 2012, 2013],
    'Report Page': [1, 2, 1],
    'Value1': [10, 15, 20]
}

df_uneven = pd.DataFrame(data_uneven)
print("\n包含页数不一致的原始数据框:")
print(df_uneven)

df_result_uneven = df_uneven.pivot(index=['FIRM', 'YEAR'], columns='Report Page', values='Value1') \
                            .add_prefix('Value1_Page') \
                            .reset_index() \
                            .rename_axis(None, axis=1)

print("\n处理页数不一致后的结果:")
print(df_result_uneven)
登录后复制

输出结果:

包含页数不一致的原始数据框:
  FIRM  YEAR  Report Page  Value1
0    A  2012            1       10
1    A  2012            2       15
2    B  2013            1       20

处理页数不一致后的结果:
  FIRM  YEAR  Value1_Page1  Value1_Page2
0    A  2012          10.0          15.0
1    B  2013          20.0           NaN
登录后复制

可以看到,对于公司 B 在 2013 年只有 Value1_Page1 而没有 Value1_Page2 的情况,Value1_Page2 列自动填充了 NaN。您可以根据需要使用 fillna() 方法来处理这些 NaN 值,例如将其替换为 0 或其他默认值。

# 示例: 填充 NaN
df_result_uneven_filled = df_result_uneven.fillna(0)
print("\n填充 NaN 后的结果:")
print(df_result_uneven_filled)
登录后复制

输出结果:

填充 NaN 后的结果:
  FIRM  YEAR  Value1_Page1  Value1_Page2
0    A  2012          10.0          15.0
1    B  2013          20.0           0.0
登录后复制

6. 注意事项与总结

  • pivot 与 pivot_table 的区别 pivot 要求 index 和 columns 的组合必须是唯一的,如果存在重复组合,它将报错。而 pivot_table 在遇到重复组合时,可以应用聚合函数(如 sum, mean 等)来处理冲突,因此 pivot_table 更为通用和灵活。在本例中,FIRM, YEAR, Report Page 的组合是唯一的,所以 pivot 是合适的选择。
  • 多值列处理: 如果需要将多个 values 列进行重塑,pivot 同样支持。只需将 values 参数传入一个列表即可。
  • 性能考量: 对于非常大的数据集,pivot 操作可能会消耗较多的内存和计算资源。在处理大数据时,应注意优化内存使用或考虑其他分布式计算框架。

通过本教程,您应该已经掌握了如何使用 Pandas 的 pivot 函数及其辅助方法,将长格式的行数据转换为宽格式的列数据,有效地重塑数据以满足特定的分析和报告需求。这种数据转换能力是 Pandas 在数据科学领域中不可或缺的一部分。

以上就是将行数据转换为列:Pandas pivot 方法详解的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号