讲师中心 微信公众号

工具推荐

视频效率加速
AniShort
AniShort

AI短剧协同制作平台

剧云
剧云

中文短剧剧本制作平台

立刻MV
立刻MV

低成本歌曲 MV、视频生成

Laper
Laper

剧本策划到成片协作

如何高效更新Pandas DataFrame中的指定观测值(避免内存爆炸)

星伟同学_7781

星伟同学_7781

发布时间:2026-02-20 17:16:04

|

417人浏览过

|

来源于php中文网

原创

如何高效更新Pandas DataFrame中的指定观测值(避免内存爆炸)

本文详解在pandas中安全、高效地替换列中特定字符串值的方法,重点解决因数据类型不当导致的“内存分配失败”问题,并推荐使用分类数据类型(categorical dtype)实现低内存开销的原地更新。

本文详解在pandas中安全、高效地替换列中特定字符串值的方法,重点解决因数据类型不当导致的“内存分配失败”问题,并推荐使用分类数据类型(categorical dtype)实现低内存开销的原地更新。

在Pandas中更新某列的特定值(例如将所有 "BB2" 替换为 "BB"),看似只需一行 df.loc[df["RATING"] == "BB2", "RATING"] = "BB" 即可完成。但若执行时抛出类似 Unable to allocate 1.41 TiB for an array... 的错误,问题通常不在于逻辑错误,而在于底层数据类型引发的隐式内存膨胀——尤其当 RATING 列是 object 类型且含大量重复字符串时,Pandas 在某些操作链(如链式索引或内部布尔索引广播)中可能触发临时大数组分配,导致内存峰值远超实际所需。

根本解决方案是:将高重复率的字符串列转换为 category 类型。该类型将字符串映射为整数编码(codes)+ 类别列表(categories),大幅压缩内存占用(通常节省 50%–90%),同时支持高效的类别级操作,且完全兼容 .loc 和 .cat 方法。

以下为推荐的三步实践流程:

✅ 步骤 1:转换为 category 类型

# 将 RATING 列转为分类类型(自动提取唯一值作为 categories)
df["RATING"] = df["RATING"].astype("category")

? 提示:astype("category") 是轻量操作,不会复制原始字符串数据,仅构建编码映射。

✅ 步骤 2:通过 cat.rename_categories() 安全重命名

# 直接修改类别名称("BB2" → "BB"),所有对应编码自动生效
df["RATING"] = df["RATING"].cat.rename_categories({"BB2": "BB"})

⚠️ 注意:rename_categories() 要求字典键必须严格匹配现有类别名(区分大小写、空格)。若不确定当前类别,可先检查:print(df["RATING"].cat.categories)。

✅ 步骤 3(可选):验证与优化

# 查看更新结果
print(df)
# 输出:
#    RATING
# 1   BB1
# 2    BB   # ← 已更新
# 3   BB1
# 4   BB3

# 检查内存使用(对比转换前后)
print(df.memory_usage(deep=True))

? 为什么不用 df.loc[...] = ...?

虽然 df.loc[df["RATING"]=="BB2", "RATING"] = "BB" 在小数据上可行,但在大数据集或 object 类型列上,Pandas 可能因内部字符串比较、布尔索引广播或视图/副本机制触发意外的内存分配。而 category 类型将比较转化为整数运算,彻底规避字符串处理开销。

? 关键注意事项

  • 仅对高重复率字符串列启用 category:若每行值几乎唯一(如用户ID、时间戳字符串),category 反而增加开销。
  • 避免混合类型操作:更新后若需与其他非 category 列计算,注意类型兼容性;必要时可用 .astype(str) 转回字符串。
  • 链式赋值警告:始终使用 df[col] = ... 而非 df[col].something = ...,确保操作作用于原DataFrame。

掌握 category 类型不仅是解决本次报错的钥匙,更是处理大规模文本特征(如评级、状态码、地区标签)的性能基石。在数据加载阶段即规划 dtype,可让后续清洗、分组、建模全流程更稳健高效。

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

热门AI工具

更多
Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

花生AI
花生AI Hot

一款AI视频创作工具,主要用于B站推出的AI视频创作工具,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

相关专题

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

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

360

2025.12.04

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

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

292

2026.01.31

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

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

931

2026.04.13

python中print函数的用法
python中print函数的用法

python中print函数的语法是“print(value1, value2, ..., sep=' ', end=' ', file=sys.stdout, flush=False)”。本专题为大家提供print相关的文章、下载、课程内容,供大家免费下载体验。

3440

2023.09.27

python print用法与作用
python print用法与作用

本专题整合了python print的用法、作用、函数功能相关内容,阅读专题下面的文章了解更多详细教程。

269

2026.02.03

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

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

3491

2023.10.31

php数据类型
php数据类型

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

574

2025.10.31

c语言 数据类型
c语言 数据类型

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

582

2026.02.12

Go-Zero API开发使用教程
Go-Zero API开发使用教程

本专题聚焦Go-Zero的API开发,拆解.api描述文件的语法结构、路由与分组定义、参数绑定与校验规则,说明统一响应体、JWT鉴权、跨域与接口文档生成的落地方式,并给出网关聚合与接口测试的实践建议。

0

2026.11.12

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn