0

0

Python中如何优化DataFrame内存占用—astype类型转换技巧详解

星夢妙者

星夢妙者

发布时间:2025-07-09 08:42:02

|

627人浏览过

|

来源于php中文网

原创

python中处理大规模数据时,通过合理使用astype()进行类型转换可以减少dataframe的内存消耗。首先查看当前内存使用情况,用df.info(memory_usage=true)或df.memory_usage(deep=true)获取各列数据类型和内存占用;接着选择合适的数据类型,如int8替代int64、float32替代float64、类别型数据使用category类型;然后可通过循环批量转换多个列的类型;最后注意缺失值处理、避免盲目降级以及合理使用category类型。这些方法能有效优化内存占用,提升处理效率。

Python中如何优化DataFrame内存占用—astype类型转换技巧详解

在Python中处理大规模数据时,DataFrame的内存占用常常成为瓶颈。Pandas默认为每一列分配相对宽松的数据类型(比如float64、int64),虽然通用性强,但不一定最节省内存。通过合理使用astype()进行类型转换,可以有效减少内存消耗,提升处理效率。

Python中如何优化DataFrame内存占用—astype类型转换技巧详解

查看当前内存使用情况

优化的第一步是了解当前DataFrame各列的内存使用情况。你可以用如下方法:

df.info(memory_usage=True)

或者查看具体数值:

立即学习Python免费学习笔记(深入)”;

Python中如何优化DataFrame内存占用—astype类型转换技巧详解
df.memory_usage(deep=True)

这些信息会告诉你每列的数据类型和实际占用内存大小。有了这些基础数据,就可以开始有针对性地优化了。

选择合适的数据类型

不同类型的数据所占内存不同,例如:

Python中如何优化DataFrame内存占用—astype类型转换技巧详解
  • int64 占8字节,而 int8 只占1字节;
  • float64 同样占8字节,如果精度要求不高,可以用 float32
  • 类别型数据(如性别、状态)适合用 category 类型存储,尤其当唯一值较少时,能大幅压缩内存。

举个例子:如果你有一列“用户等级”,取值范围是1~5,用int64就明显浪费空间,改用int8即可。

Multiavatar
Multiavatar

Multiavatar是一个免费开源的多元文化头像生成器,可以生成高达120亿个虚拟头像

下载
df['level'] = df['level'].astype('int8')

批量转换多个列的类型

如果你有多列需要调整类型,可以写个小循环批量操作:

cols_to_downcast = ['age', 'score', 'count']
for col in cols_to_downcast:
    df[col] = df[col].astype('int16')  # 或其他合适的类型

这种方式适用于结构化数据,尤其是从数据库读入后未做类型优化的情况。

注意事项与常见问题

  • 缺失值处理:某些类型不支持NaN(如int系列),如果列中有缺失值,直接转换会报错。可以先填充或删除缺失值,再转换。

    df['col'] = df['col'].fillna(0).astype('int8')
  • 不要盲目降级:比如把本来可能有大整数的列转成int8,会导致溢出错误。务必确认数据范围后再做转换。

  • category类型的适用场景:类别种类越少,用category的效果越好。如果某一列几乎都是唯一值,就不适合用这个类型。

基本上就这些。内存优化是个细节活,不能一概而论,但只要掌握基本思路和工具,就能在大多数情况下显著减少DataFrame的内存占用。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

71

2025.12.04

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

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

1

2026.01.31

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

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

310

2023.10.31

php数据类型
php数据类型

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

222

2025.10.31

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

483

2023.08.02

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

545

2024.08.29

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

113

2025.08.29

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

200

2025.08.29

2026赚钱平台入口大全
2026赚钱平台入口大全

2026年最新赚钱平台入口汇总,涵盖任务众包、内容创作、电商运营、技能变现等多类正规渠道,助你轻松开启副业增收之路。阅读专题下面的文章了解更多详细内容。

33

2026.01.31

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
进程与SOCKET
进程与SOCKET

共6课时 | 0.4万人学习

UNI-APP开发(仿饿了么)
UNI-APP开发(仿饿了么)

共32课时 | 8.8万人学习

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

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