0

0

如何在不将CSV文件全部加载到内存的情况下高效计算用户年龄中位数

碧海醫心

碧海醫心

发布时间:2026-01-27 11:09:10

|

323人浏览过

|

来源于php中文网

原创

如何在不将CSV文件全部加载到内存的情况下高效计算用户年龄中位数

本文介绍如何使用pythoncsv.dictreader流式读取tsv文件,避免内存占用过高,同时解决“_csv.reader对象不可索引”错误,正确计算用户年龄中位数。

在处理大型CSV/TSV文件时,为节省内存,我们通常希望逐行读取、即时计算,而非一次性加载全部数据(如pandas.read_csv()或list(csv.reader()))。但初学者常误用csv.reader——它返回的是按列索引的列表(如row[2]),而代码中却尝试像字典一样用键名访问(如row['birth_year']),从而触发 TypeError: '_csv.reader' object is not subscriptable 错误。

根本原因在于:csv.reader 生成的每一行是 list 类型(例如 ['paul', 'henry', '2019']),不支持 row['birth_year'] 这种键访问;而 csv.DictReader 才会将首行自动解析为字段名,并为后续每行返回一个 dict(如 {'first': 'paul', 'last': 'henry', 'birth_year': '2019'}),真正支持键值访问。

✅ 正确做法是改用 csv.DictReader,并配合生成器表达式实现内存友好的流式计算:

import csv
from statistics import median

def median_age(filename):
    with open(filename, 'r', encoding='utf-8') as file:
        reader = csv.DictReader(file, delimiter='\t')

        # 使用生成器:逐行读取,即时计算年龄,不构建完整列表
        ages = (2024 - int(row['birth_year']) for row in reader)

        return median(ages)

? 关键要点说明:

Cardify卡片工坊
Cardify卡片工坊

使用Markdown一键生成精美的小红书知识卡片

下载
  • ✅ csv.DictReader 自动将第一行作为字段名(header),无需手动跳过(next(reader) 不再需要);
  • ✅ 生成器 (2024 - int(...)) for row in reader 不会缓存所有年龄值,仅维持迭代状态,内存占用恒定(O(1));
  • ✅ 建议显式指定 encoding='utf-8' 避免中文或特殊字符报错;
  • ⚠️ 确保输入文件中存在名为 'birth_year' 的列(大小写敏感),否则抛出 KeyError;可添加异常处理增强鲁棒性:
# 可选:增加字段校验与容错
if 'birth_year' not in reader.fieldnames:
    raise ValueError(f"Column 'birth_year' not found in header: {reader.fieldnames}")

? 示例运行(以制表符分隔的 data.csv):

first   last    birth_year
paul    henry   2019
bill    thompson    1995
mary    allen   2003
jennifer    davis   2015
liz morgan  1999

调用 median_age('data.csv') 将依次计算年龄 [5, 29, 21, 9, 25],中位数为 21(已排序后取中间值)。

总结:流式处理大数据CSV的核心是「用对工具」——DictReader 提供语义化访问,生成器确保低内存开销,二者结合即可安全、高效、可维护地完成年龄统计任务。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

82

2025.12.04

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

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

34

2026.01.31

string转int
string转int

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

1051

2023.08.02

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

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

615

2024.08.29

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

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

335

2025.08.29

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

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

235

2025.08.29

TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

42

2026.03.13

Python异步编程与Asyncio高并发应用实践
Python异步编程与Asyncio高并发应用实践

本专题围绕 Python 异步编程模型展开,深入讲解 Asyncio 框架的核心原理与应用实践。内容包括事件循环机制、协程任务调度、异步 IO 处理以及并发任务管理策略。通过构建高并发网络请求与异步数据处理案例,帮助开发者掌握 Python 在高并发场景中的高效开发方法,并提升系统资源利用率与整体运行性能。

79

2026.03.12

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

234

2026.03.11

热门下载

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

精品课程

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

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

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

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