
本教程详细介绍了如何使用 pandas 的 `pivot` 方法将数据框中按行存储的页面级信息转换为按列展示的报告级汇总数据。通过指定索引、列和值参数,结合 `add_prefix`、`reset_index` 和 `rename_axis` 等辅助操作,实现数据重塑,将不同页码的值转换为独立的列,从而高效地聚合和展示复杂的数据结构,同时处理不同报告页数不一致的情况。
在数据分析和处理中,我们经常会遇到需要将“长格式”数据(即某些属性的值作为行来存储)转换为“宽格式”数据(即这些属性的值作为新的列来存储)的场景。例如,当一份报告包含多个页面的数据,并且每个页面的信息都作为独立的一行存储时,我们可能希望将同一报告(由公司和年份标识)的所有页面数据聚合到一行中,并为每个页面创建一个独立的列。Pandas 库提供了强大的 pivot 函数来高效地完成这类数据重塑任务。
假设我们有一份关于公司年度报告的数据,其中包含了公司(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
pandas.pivot 函数是实现这种行转列操作的核心工具。它接收三个关键参数:
针对我们的需求,我们可以这样使用 pivot:
基本 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 仍是索引。
为了达到最终的期望格式,我们还需要进行一系列的辅助操作:
为了使列名更具描述性,我们可以使用 add_prefix() 方法为新生成的列名添加一个前缀。
电子手机配件网站源码是一个响应式的织梦网站模板,软件兼容主流浏览器,且可以在PC端和手机端中进行浏览。模板包含安装说明,并包含测试数据。本模板基于DEDECms 5.7 UTF-8设计,需要GBK版本的请自己转换。模板安装方法:1、下载最新的织梦dedecms5.7 UTF-8版本。2、解压下载的织梦安装包,得到docs和uploads两个文件夹,请将uploads里面的所有文件和文件夹上传到你的
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 25pivot 操作会将 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
在 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
将上述所有步骤合并,可以得到简洁高效的解决方案:
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)教程中提到“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
通过本教程,您应该已经掌握了如何使用 Pandas 的 pivot 函数及其辅助方法,将长格式的行数据转换为宽格式的列数据,有效地重塑数据以满足特定的分析和报告需求。这种数据转换能力是 Pandas 在数据科学领域中不可或缺的一部分。
以上就是将行数据转换为列:Pandas pivot 方法详解的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号