0

0

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

聖光之護

聖光之護

发布时间:2026-01-23 11:39:01

|

356人浏览过

|

来源于php中文网

原创

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

本文介绍一种无需显式双重 for 循环的优雅方式,利用 pandas 的链式 groupby 和 apply 方法,将 dataframe 按多个列分组并转换为深度嵌套字典(如 {level1: {level2: [{record}, ...]}})。

在数据处理中,常需将扁平化的 DataFrame 转换为结构化嵌套字典,以适配 API 请求、配置生成或前端树形渲染等场景。传统双重 for 循环虽直观,但代码冗长、可读性差且难以扩展。更优解是充分利用 pandas 的分组聚合能力——通过嵌套 groupby().apply() 实现声明式层级构建。

核心思路是:外层按第一级键(如 'col1')分组 → 内层对每个子组再按第二级键(如 'col2')分组 → 对最内层子组调用 to_dict(orient='records') 生成记录列表。配合 .to_dict('index') 可自动将外层分组索引转为字典键。

以下为推荐实现(简洁、高效、可读性强):

import pandas as pd

# 构造示例数据
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},
])

# 一行式嵌套转换(推荐)
result = a.groupby('col1').apply(
    lambda x: x.groupby('col2').apply(
        lambda y: y.to_dict('records')
    )
).to_dict('index')

优势说明

塔猫ChatPPT
塔猫ChatPPT

塔猫官网提供AI一键生成 PPT的智能工具,帮助您快速制作出专业的PPT。塔猫ChatPPT让您的PPT制作更加简单高效。

下载
  • 零显式循环:完全避免 for 循环,逻辑聚焦于“按什么分组”和“每组转什么”,语义清晰;
  • 天然层级映射:groupby('col1').apply(...) 输出 Series,索引为 'col1' 值;其值为另一 Series(索引为 'col2'),再经 to_dict('index') 即得两层嵌套字典;
  • 可扩展性强:若需三级嵌套(如再按 'category' 分组),只需追加 .groupby('category').apply(...) 即可;
  • 性能良好:基于 pandas 底层优化,比纯 Python 循环更高效,尤其在大数据集上。

⚠️ 注意事项

  • to_dict('index') 是关键收尾步骤,它将外层 groupby().apply() 返回的 Series 自动转换为以 'col1' 值为键的字典;若省略,会得到一个带 MultiIndex 的 Series,不符合预期结构;
  • y.to_dict('records') 确保每组内所有行转为字典列表(保留原始字段,含分组列);若不需 'col1'/'col2' 字段,可在 y 上先 drop(['col1','col2'], axis=1);
  • 若分组列存在缺失值(NaN),需提前处理(如 dropna=True 在 groupby 中),否则 NaN 会作为键导致不可哈希错误。

总结:通过 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相关的各种文章、以及下载和课程。

764

2023.07.25

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

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

679

2023.07.31

python教程
python教程

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

1365

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++ 中的高级模板编程与元编程技术,涵盖模板特化、SFINAE、模板递归、类型萃取、编译时常量与计算、C++17 的折叠表达式与变长模板参数等。通过多个实际示例,帮助开发者掌握 如何利用 C++ 模板机制编写高效、可扩展的通用代码,并提升代码的灵活性与性能。

2

2026.01.23

热门下载

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

精品课程

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

共4课时 | 14.2万人学习

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号