如何为 DataFrame 创建映射至嵌套列表层级的分类列

阿墨同学_6159

阿墨同学_6159

2026-05-10

529人浏览

原创

如何为 DataFrame 创建映射至嵌套列表层级的分类列

本文介绍一种高效、向量化的方法,将 dataframe 中某列的值依据其所属的预定义列表组(如 tier1、tier2),映射为对应的层级编号(如 1、2),并生成新分类列,避免显式循环,适用于数百行数据与十余个分组场景。

本文介绍一种高效、向量化的方法,将 dataframe 中某列的值依据其所属的预定义列表组(如 tier1、tier2),映射为对应的层级编号(如 1、2),并生成新分类列,避免显式循环,适用于数百行数据与十余个分组场景。

在 Pandas 数据处理中,常需根据成员归属关系为记录打上分类标签(例如按人员名单划分“Tier 1”“Tier 2”)。若直接使用 iterrows() 遍历 + 嵌套 in 判断,不仅代码冗长,且对 300+ 行、11 个列表的规模而言性能低下。更优解是构建反向映射字典(name → tier_id),再通过 .map() 实现 O(1) 查找,完全向量化、简洁且高效。

核心思路是:将多个列表(如 tier1, tier2)扁平化为一个字典,键为每个姓名,值为其所属层级编号。推荐使用字典推导式一次性完成构建:

import pandas as pd

# 示例数据
d = {'name': ['Tom', 'Patrick', 'Lamar'], 'rating': [100, 97, 95]}
df = pd.DataFrame(d)

# 定义层级列表(实际中可有 tier1 ~ tier11)
tier1 = ['Tom', 'Patrick']
tier2 = ['Lamar']

# ✅ 构建 name → tier_id 映射字典(关键步骤)
tier_dict = {
    name: tier_id 
    for tier_id, tier_list in enumerate([tier1, tier2], start=1)
    for name in tier_list
}
# 结果:{'Tom': 1, 'Patrick': 1, 'Lamar': 2}

# ✅ 向量化赋值:利用 map 自动对齐并填充 NaN(若 name 不在任何 tier 中)
df['tier'] = df['name'].map(tier_dict)

执行后,df 即包含正确分级的 'tier' 列:

      name  rating  tier
0      Tom     100     1
1  Patrick      97     1
2    Lamar      95     2

⚠️ 注意事项:

  • 若某 name 不属于任何 tier 列表,.map() 默认返回 NaN,可后续用 fillna() 处理(如 df['tier'] = df['name'].map(tier_dict).fillna(0));
  • 确保各 tier 列表无重复姓名,否则后出现的映射会覆盖前者(因字典键唯一);
  • 如需动态扩展(如 11 个 tier),只需将 [tier1, tier2] 替换为 [tier1, tier2, ..., tier11] 或 globals() 动态获取(但更推荐预先组织为列表或配置字典);
  • 此方法时间复杂度为 O(N),远优于双重循环的 O(N×M),尤其适合中等规模数据。

总结:用字典推导式构建反向索引 + .map() 向量化映射,是解决“列表归属→标签赋值”类问题的 Pythonic 标准方案——简洁、高效、可读性强,且天然兼容 Pandas 生态。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

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

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

2025.12.04

320

15

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

2026.01.31

252

21

Python数据分析与Pandas高级实战
Python数据分析与Pandas高级实战

本专题围绕 Python 数据分析展开,系统讲解 Pandas 的高级用法,包括数据清洗、透视表、时间序列分析以及多表合并与分组操作。通过实战案例,帮助开发者掌握高效处理与分析数据的方法,提高数据处理效率与分析能力。

2026.04.13

491

25

PixTV官网入口地址合集
PixTV官网入口地址合集

本专题汇总了 PixTV AI 一站式视频创作平台的官方入口与使用教程。无需下载软件,浏览器直接访问即可使用。平台将剧本、图像、视频、声音与剪辑整合在“无限画布”中,接入 GPT Image 2.5、Seedance 2.5 等头部模型。本专题整理了从新建画布、角色锚定、分镜拆分到视频生成与导出的完整操作指南,助你快速上手 AI 短剧与漫剧创作。

2026.10.10

80

15

Kratos框架HTTP与gRPC服务开发教程
Kratos框架HTTP与gRPC服务开发教程

本专题围绕Kratos框架双协议服务开发,涵盖HTTP路由与处理器编写、参数获取、gRPC服务实现与客户端调用、metadata上下文传递、encoding编解码注册、统一响应封装、超时控制与流式响应实现方法。

2026.10.10

40

15

Kratos框架Protobuf接口定义与代码生成合集
Kratos框架Protobuf接口定义与代码生成合集

本专题讲解Kratos框架接口定义体系,涵盖proto编写规范、proto add/client/server生成命令、http注解路由、validate校验、OpenAPI文档生成、跨服务proto复用与兼容性设计。

2026.10.10

20

15

C++虚函数怎么定义和调用
C++虚函数怎么定义和调用

C++虚函数是实现运行时多态的重要机制。本专题从virtual关键字的基本用法入手,介绍基类与派生类之间的函数重写、基类指针调用派生类方法,以及动态绑定的执行过程,帮助初学者掌握虚函数的核心语法。

2026.10.10

20

26

C++类与对象的封装方法教程
C++类与对象的封装方法教程

C++封装是面向对象编程的核心特性之一,通过类将数据与操作数据的函数组织在一起,并利用访问权限控制外部访问。本专题介绍类的定义、成员变量、成员函数以及public、private和protected的使用方法,帮助初学者掌握封装的基本原理。

2026.10.10

20

32

C++构造函数定义与调用方法
C++构造函数定义与调用方法

C++构造函数用于初始化类对象,是面向对象编程的重要基础。本专题从构造函数的定义、声明和调用入手,介绍默认构造函数、带参数构造函数、拷贝构造函数及成员初始化列表,帮助初学者掌握对象创建与初始化的基本方法。

2026.10.10

20

16

热门下载

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

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习