0

0

Pandas DataFrame:高效统计每列唯一值并转换为字典

霞舞

霞舞

发布时间:2025-10-27 14:13:25

|

785人浏览过

|

来源于php中文网

原创

Pandas DataFrame:高效统计每列唯一值并转换为字典

本教程旨在介绍如何使用 pandas 库高效地统计 dataframe 中每一列的唯一值及其出现频率,并将结果整合成一个嵌套字典结构。我们将探讨一种简洁且避免在 dataframe 层面进行复杂迭代的方法,而是利用 pandas series 的内置优化功能,实现将列名映射到其对应唯一值计数字典的目标,从而优化数据处理和分析流程。

引言

在数据分析和处理中,我们经常需要了解 DataFrame 中各列数据的分布情况,特别是每列中不同值的出现频率。例如,在一个包含用户行为数据的 DataFrame 中,我们可能需要统计每个产品ID或事件类型出现的次数。将这些统计结果以嵌套字典的形式组织起来,即外层字典的键是列名,内层字典的键是列中的唯一值,值是其出现次数,是一种非常清晰和实用的数据结构。

问题阐述与目标

我们的目标是给定一个 Pandas DataFrame,生成一个如下结构的字典:

{'列名1': {'唯一值A': 计数A, '唯一值B': 计数B, ...},
 '列名2': {'唯一值X': 计数X, '唯一值Y': 计数Y, ...}}

例如,对于以下 DataFrame:

import pandas as pd

data = {'Col1': [1, 2, 2, 3, 1],
        'Col2': ['A', 'B', 'B', 'A', 'C']}
df = pd.DataFrame(data)

我们期望得到:

{'Col1': {1: 2, 2: 2, 3: 1},
 'Col2': {'A': 2, 'B': 2, 'C': 1}}

在实现过程中,我们希望采用一种高效且简洁的方式,尽量避免显式的行级别循环或复杂的 DataFrame 转换操作(如 apply 或 agg)。

解决方案详解

Pandas 提供了一种非常直接且高效的方法来解决这个问题,它利用了字典推导式(dictionary comprehension)和 Series 的 value_counts() 方法。

MagickPen
MagickPen

在线AI英语写作助手,像魔术师一样在几秒钟内写出任何东西。

下载

核心思想是:

  1. 遍历 DataFrame 的每一列(通过列名)。
  2. 对于每一列,将其视为一个 Pandas Series。
  3. 使用 Series 的 value_counts() 方法来统计该 Series 中每个唯一值的出现频率。
  4. 将 value_counts() 的结果(一个 Series)转换为字典 (.to_dict())。
  5. 将列名作为键,将转换后的字典作为值,构建最终的嵌套字典。

value_counts() 方法是 Pandas Series 提供的一个强大功能,它返回一个包含唯一值及其对应计数的 Series,默认按计数降序排列。此方法在底层经过高度优化,效率非常高。

示例代码

下面是实现上述目标的完整代码示例:

import pandas as pd

# 1. 准备示例数据
data = {'Col1': [1, 2, 2, 3, 1],
        'Col2': ['A', 'B', 'B', 'A', 'C'],
        'Col3': [True, False, True, None, True]} # 增加一列包含不同类型和缺失值

df = pd.DataFrame(data)

print("原始 DataFrame:")
print(df)
print("-" * 30)

# 2. 生成唯一值计数字典的解决方案
# 使用字典推导式遍历DataFrame的每一列
result_dict = {col: df[col].value_counts().to_dict() for col in df.columns}

print("\n生成的唯一值计数字典:")
print(result_dict)
print("-" * 30)

# 3. 验证 Col3 的结果,注意 None/NaN 的处理
# 默认情况下,value_counts() 会忽略 NaN 值
result_dict_with_nan = {col: df[col].value_counts(dropna=False).to_dict() for col in df.columns}
print("\n包含 NaN 计数的唯一值计数字典 (Col3):")
print(result_dict_with_nan['Col3'])

输出结果:

原始 DataFrame:
   Col1 Col2   Col3
0     1    A   True
1     2    B  False
2     2    B   True
3     3    A   None
4     1    C   True
------------------------------

生成的唯一值计数字典:
{'Col1': {1: 2, 2: 2, 3: 1}, 'Col2': {'A': 2, 'B': 2, 'C': 1}, 'Col3': {True: 3, False: 1}}
------------------------------

包含 NaN 计数的唯一值计数字典 (Col3):
{True: 3, False: 1, None: 1}

方法优势

  1. 简洁性与可读性: 单行字典推导式使得代码极其简洁,易于理解。
  2. 效率高: 尽管字典推导式涉及对列名的迭代,但每个 df[col].value_counts() 操作都是在 Pandas 内部使用 C 语言实现的,高度优化,因此整体性能非常出色,尤其对于大数据集。它避免了 Python 层面低效的逐行或逐元素循环。
  3. 避免中间状态: 直接生成目标字典结构,不需要创建额外的中间 DataFrame 或 Series,减少内存消耗和计算步骤。
  4. 灵活性: value_counts() 方法本身提供了 normalize(计算频率而非计数)、sort(是否排序)、ascending(升序或降序)、bins(分箱)和 dropna(是否包含 NaN 值)等参数,可以根据具体需求进行调整。

注意事项

  • 缺失值(NaN)处理: value_counts() 默认会忽略 NaN 值。如果需要将 NaN 也计入统计,可以设置 dropna=False,如示例代码中所示。
  • 数据类型: value_counts() 适用于各种数据类型(数值、字符串、布尔值等)。
  • 大数据量DataFrame: 对于拥有极多列(例如数千列)的 DataFrame,字典推导式中的列迭代可能会带来轻微的开销。但在绝大多数实际应用场景中,这种方法都是性能和简洁性的最佳平衡点。

总结

通过利用 Pandas Series 的 value_counts() 方法和 Python 的字典推导式,我们可以高效且优雅地统计 DataFrame 中每一列的唯一值及其频率,并将其组织成一个结构清晰的嵌套字典。这种方法不仅代码简洁,而且在性能上表现出色,是处理此类数据聚合任务的推荐实践。理解并掌握这种技巧,将有助于提升您在 Pandas 数据处理中的效率和代码质量。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

65

2025.12.04

数据类型有哪几种
数据类型有哪几种

数据类型有整型、浮点型、字符型、字符串型、布尔型、数组、结构体和枚举等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

309

2023.10.31

php数据类型
php数据类型

本专题整合了php数据类型相关内容,阅读专题下面的文章了解更多详细内容。

222

2025.10.31

sort排序函数用法
sort排序函数用法

sort排序函数的用法:1、对列表进行排序,默认情况下,sort函数按升序排序,因此最终输出的结果是按从小到大的顺序排列的;2、对元组进行排序,默认情况下,sort函数按元素的大小进行排序,因此最终输出的结果是按从小到大的顺序排列的;3、对字典进行排序,由于字典是无序的,因此排序后的结果仍然是原来的字典,使用一个lambda表达式作为key参数的值,用于指定排序的依据。

391

2023.09.04

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

298

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

212

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1502

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

624

2023.11.24

java入门学习合集
java入门学习合集

本专题整合了java入门学习指南、初学者项目实战、入门到精通等等内容,阅读专题下面的文章了解更多详细学习方法。

1

2026.01.29

热门下载

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

精品课程

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

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.7万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

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

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