0

0

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

碧海醫心

碧海醫心

发布时间:2026-02-20 20:48:01

|

484人浏览过

|

来源于php中文网

原创

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

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

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

在Pandas中更新DataFrame中特定观测值(如将所有"BB2"替换为"BB")看似简单,但若直接对大型字符串列执行df.loc[df["RATING"]=="BB2", "RATING"] = "BB",极可能触发Unable to allocate X TiB for an array...这类内存错误。该错误并非代码逻辑错误,而是源于Pandas在布尔索引过程中对object类型字符串列的隐式内存放大行为:当DataFrame规模较大(例如数十万行以上)且字符串列未优化时,临时布尔掩码和副本操作会尝试分配远超实际所需的空间——尤其在旧版Pandas或内存受限环境中尤为常见。

根本原因在于:默认的object dtype将每个字符串存储为独立Python对象指针,缺乏内存连续性与压缩能力;而布尔索引df["RATING"]=="BB2"需遍历全部字符串并构建全量布尔数组,极易引发内存峰值。

推荐解决方案:使用 category 数据类型

category 是专为重复性离散值(如评级、状态、类别标签)设计的高效存储类型。它将字符串映射为整数编码(codes),底层仅存储唯一类别(categories)一次,大幅降低内存占用(通常可减少50%–90%),同时保持语义清晰和操作便捷。

文希AI写作
文希AI写作

AI论文写作平台

下载

以下是完整实践步骤:

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

# 2. 通过cat.rename_categories安全重命名指定类别(不改变其他值)
df["RATING"] = df["RATING"].cat.rename_categories({"BB2": "BB"})

# 验证结果
print(df)
#    RATING
# 1   BB1
# 2    BB   ← 已更新
# 3   BB1
# 4   BB3

⚠️ 注意事项:

  • rename_categories() 仅修改类别标签,不会影响已有的编码映射关系,是安全的原地更新;
  • 若需新增类别(如将不存在的"BB4"映射为"BB"),应先用 cat.add_categories() 扩展类别池;
  • 转换为category后,仍可正常使用.str方法(需先.astype(str))、排序、分组等操作,兼容性良好;
  • 对于超大规模数据(千万行+),建议在读取阶段即指定dtype:pd.read_csv(..., dtype={"RATING": "category"}),从源头规避内存问题。

? 补充技巧:批量替换多个值 若需同时更新多个类别,可传入字典:

df["RATING"] = df["RATING"].cat.rename_categories({
    "BB1": "B+", 
    "BB2": "BB", 
    "BB3": "B-"
})

总结:面对字符串列的批量更新需求,优先考虑category dtype而非盲目使用.loc赋值。它不仅规避了内存灾难,还提升了后续计算性能(如groupby、value_counts)。对于新手而言,养成对高重复率文本列主动设为category的习惯,是写出健壮、可扩展Pandas代码的关键一步。

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

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

75

2025.12.04

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

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

7

2026.01.31

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

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

311

2023.10.31

php数据类型
php数据类型

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

223

2025.10.31

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

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

53

2026.02.12

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

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

594

2023.08.03

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

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

217

2023.09.04

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

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

1555

2023.10.24

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

796

2026.02.13

热门下载

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

精品课程

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

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