0

0

如何将 DataFrame 高效转换为多层嵌套字典

心靈之曲

心靈之曲

发布时间:2026-01-23 13:42:01

|

179人浏览过

|

来源于php中文网

原创

如何将 DataFrame 高效转换为多层嵌套字典

本文介绍使用 pandas 的链式 `groupby` 与 `apply` 方法,无需显式循环即可将 dataframe 转换为任意深度(如两层)的嵌套字典,显著提升代码简洁性与可读性。

在数据处理中,常需将扁平化的 DataFrame 按多个层级键(如 'col1' → 'col2')组织为嵌套字典结构,便于后续 JSON 序列化、API 响应构造或配置管理。传统双层 for 循环虽直观,但冗长且不易维护。Pandas 提供了更优雅的向量化方案:嵌套 groupby().apply() 配合 to_dict('records')

核心思路是逐级分组并聚合:

  • 先按第一级键(如 'col1')分组;
  • 对每个子组再按第二级键(如 'col2')分组;
  • 将最内层子组直接转为记录列表(to_dict('records'));
  • 最终调用 .to_dict('index') 将外层 Series 转为字典。

以下是完整实现:

import pandas as pd
from pprint import pprint

# 构造示例数据
a = pd.DataFrame([
    {'col1': 'A', 'col2': 'Person 1', 'height': 1, 'weight': 10},
    {'col1': 'A', 'col2': 'Person 1', 'height': 2, 'weight': 20},
    {'col1': 'A', 'col2': 'Person 1', 'height': 3, 'weight': 30},
    {'col1': 'A', 'col2': 'Person 2', 'height': 4, 'weight': 40},
    {'col1': 'A', 'col2': 'Person 2', 'height': 5, 'weight': 50},
    {'col1': 'A', 'col2': 'Person 2', 'height': 6, 'weight': 60},
    {'col1': 'B', 'col2': 'Person 1', 'height': 11, 'weight': 101},
    {'col1': 'B', 'col2': 'Person 1', 'height': 21, 'weight': 201},
    {'col1': 'B', 'col2': 'Person 1', 'height': 31, 'weight': 301},
    {'col1': 'B', 'col2': 'Person 2', 'height': 41, 'weight': 401},
    {'col1': 'B', 'col2': 'Person 2', 'height': 51, 'weight': 501},
    {'col1': 'B', 'col2': 'Person 2', 'height': 61, 'weight': 601},
])

# ✅ 推荐:单行嵌套 groupby 实现两层嵌套字典
out = a.groupby('col1').apply(
    lambda x: x.groupby('col2').apply(lambda y: y.to_dict('records'))
).to_dict('index')

pprint(out)

该方法可轻松扩展至 三层及以上嵌套(例如再增加 'col3'):

青泥AI
青泥AI

青泥学术AI写作辅助平台

下载
# 三层示例(伪代码,需确保 col3 存在)
out_3level = a.groupby('col1').apply(
    lambda x: x.groupby('col2').apply(
        lambda y: y.groupby('col3').apply(lambda z: z.to_dict('records'))
    )
).to_dict('index')

⚠️ 注意事项:

  • to_dict('index') 是关键——它将 groupby().apply() 返回的 Series(索引为 'col1' 值)转为字典;若省略,结果仍是 Series 对象。
  • 内层 apply(lambda y: y.to_dict('records')) 返回的是 Series(索引为 'col2'),外层 .to_dict('index') 自动将其展开为第二层键值对
  • 性能上,此方法仍基于 Pandas 分组引擎,比纯 Python 循环更高效,尤其在中等规模数据(万行级)下优势明显。
  • 若需排除分组列(如 'col1', 'col2')以精简输出字典字段,可在 to_dict('records') 前使用 drop(columns=['col1','col2'])。

总结:利用 groupby().apply() 的函数式嵌套能力,可完全替代手动循环,写出清晰、可读、可扩展的多级嵌套字典构建逻辑,是 Pandas 高级数据重塑的典型实践。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

772

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

661

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

765

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

679

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1385

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

570

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

579

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

730

2023.08.11

c++空格相关教程合集
c++空格相关教程合集

本专题整合了c++空格相关教程,阅读专题下面的文章了解更多详细内容。

0

2026.01.23

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 15.4万人学习

Django 教程
Django 教程

共28课时 | 3.4万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

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