0

0

如何高效分块处理并排序超大CSV文件(Python教程)

霞舞

霞舞

发布时间:2026-02-09 11:46:42

|

283人浏览过

|

来源于php中文网

原创

如何高效分块处理并排序超大CSV文件(Python教程)

本文介绍使用python分块读取超大csv文件的方法,避免内存溢出,并通过fileinput模块逐行解析、按需切分字段,结合用户输入实现灵活筛选与排序。

在处理GB级CSV文件时,一次性加载全部数据到内存(如用pandas.read_csv()或csv.reader配合list())极易导致内存耗尽或程序崩溃。此时,流式分块处理是更稳健的选择——不追求“一次性分割为4个物理文件”,而是逻辑上按字段维度切分关注区域,并逐行解析、即时分类、动态聚合,从而支撑后续排序与查询。

以下是一个专业、可扩展的实现方案:

ChatPDF
ChatPDF

使用ChatPDF,您的文档将变得智能!跟你的PDF文件对话,就好像它是一个完全理解内容的人一样。

下载

✅ 推荐方案:使用 fileinput 流式读取 + 字段映射 + 动态分组

import fileinput
from collections import defaultdict
import operator

def load_and_partition_by_key(filepath, key_field=0, encoding="utf-8"):
    """
    按指定列(如 country/abv/year)流式分组数据,返回字典:{key: [(country, abv, year, expectancy), ...]}
    key_field: 0=country, 1=abv, 2=year, 3=expectancy(支持任意列作为逻辑“部分”)
    """
    partitions = defaultdict(list)

    with fileinput.input(filepath, encoding=encoding) as f:
        for line_num, line in enumerate(f, 1):
            line = line.strip()
            if not line:  # 跳过空行
                continue
            try:
                parts = line.split(",")
                # 安全提取并类型转换
                country = parts[0].strip('"\' ')
                abv = parts[1].strip('"\' ')
                year = int(parts[2])
                expectancy = float(parts[3])

                # 按用户指定字段分组(例如:按国家分组 → 一个“part”对应一个国家的所有年份记录)
                key = parts[key_field].strip('"\' ') if key_field < len(parts) else "unknown"
                partitions[key].append((country, abv, year, expectancy))

            except (ValueError, IndexError) as e:
                print(f"⚠️ 警告:第 {line_num} 行格式异常,已跳过 — {e}")
                continue

    return partitions

# 示例:按国家分4个典型部分(取前4个非空国家名),并分别排序
if __name__ == "__main__":
    data_by_country = load_and_partition_by_key("life_expectancy.csv", key_field=0)

    # 用户输入目标国家(模拟交互式筛选)
    target_countries = ["China", "India", "Brazil", "Nigeria"]  # 或由 input() 获取
    for country in target_countries:
        if country in data_by_country:
            # 对该国数据按年份升序排序
            sorted_by_year = sorted(data_by_country[country], key=operator.itemgetter(2))
            print(f"\n? {country}(共{len(sorted_by_year)}条记录):")
            for record in sorted_by_year[:3]:  # 仅显示前3年示例
                print(f"  {record[2]}年: {record[3]:.2f}岁")
        else:
            print(f"? {country} 数据未找到")

? 关键优势说明:

  • 零内存压力:fileinput 逐行读取,每行仅占用极小内存;
  • 弹性“分块”:所谓“分成4部分”,本质是按业务维度(国家/年份/缩写/指标)逻辑聚类,而非物理拆分文件;
  • 强健容错:自动跳过空行、异常行,并提示错误位置;
  • 即用即排:每个逻辑分组(如某国所有年份)可独立调用 sorted() 或 heapq.nsmallest() 实现高效排序;
  • 无缝扩展:后续可轻松接入 pandas.DataFrame(对单个分组)、写入新CSV,或构建内存索引。

⚠️ 注意事项:

  • 确保CSV无嵌套逗号(否则需用 csv 模块替代 str.split(","));
  • 若需严格按物理文件切分(如将原文件拆成4个子文件),应使用 itertools.islice 配合 open() 分段写入,但通常不推荐——它无法保证语义完整性(如切断某国连续年份);
  • 中文路径或特殊字符请始终显式指定 encoding="utf-8-sig" 防止乱码。

掌握这种“流式解析 + 键值分组 + 懒排序”范式,你不仅能应对千万行CSV,还能为构建实时分析管道打下坚实基础。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

73

2025.12.04

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

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

3

2026.01.31

包子漫画网页版入口与全集阅读指南_正版免费漫画快速访问方法
包子漫画网页版入口与全集阅读指南_正版免费漫画快速访问方法

本专题汇总了包子漫画官网和网页版入口,提供最新章节抢先看方法、正版免费阅读指南,以及稳定访问方式,帮助用户快速直达包子漫画页面,无广告畅享全集漫画内容。

47

2026.02.10

MC.JS网页版快速畅玩指南_MC.JS官网在线入口及免安装体验方法
MC.JS网页版快速畅玩指南_MC.JS官网在线入口及免安装体验方法

本专题汇总了MC.JS官网入口和网页版快速畅玩方法,提供免安装访问、不同版本(1.8.8、1.12.8)在线体验指南,以及正版网页端操作说明,帮助玩家轻松进入MC.JS世界,实现即时畅玩与高效体验。

35

2026.02.10

谷歌邮箱网页版登录与注册全指南_Gmail账号快速访问与安全操作教程
谷歌邮箱网页版登录与注册全指南_Gmail账号快速访问与安全操作教程

本专题汇总了谷歌邮箱网页版的最新登录入口和注册方法,详细提供官方账号快速访问方式、网页版操作教程及安全登录技巧,帮助用户轻松管理Gmail邮箱账户,实现高效、安全的邮箱使用体验。

27

2026.02.10

铁路12306订票与退改全攻略_高效购票与座位选取技巧
铁路12306订票与退改全攻略_高效购票与座位选取技巧

本专题全面汇总铁路12306订票、退票、改签及候补订单操作技巧,提供车厢座位分布参考、抢票攻略和高铁安检注意事项,帮助新手用户快速掌握高效购票与退改流程,提高出行效率和体验。

31

2026.02.10

TensorFlow2深度学习模型实战与优化
TensorFlow2深度学习模型实战与优化

本专题面向 AI 与数据科学开发者,系统讲解 TensorFlow 2 框架下深度学习模型的构建、训练、调优与部署。内容包括神经网络基础、卷积神经网络、循环神经网络、优化算法及模型性能提升技巧。通过实战项目演示,帮助开发者掌握从模型设计到上线的完整流程。

0

2026.02.10

Vue3组合式API与组件开发实战
Vue3组合式API与组件开发实战

本专题讲解 Vue 3 组合式 API 的核心概念与应用技巧,深入分析响应式系统、生命周期管理、组件设计与复用策略。通过完整项目案例,指导前端开发者实现高性能、结构清晰的 Vue 应用,提升开发效率与代码可维护性。

4

2026.02.10

Go语言微服务架构与gRPC实战
Go语言微服务架构与gRPC实战

本专题面向有 Go 基础的开发者,系统讲解微服务架构设计与 gRPC 的高效应用。内容涵盖服务拆分、RPC 通信、负载均衡、错误处理、服务注册与发现等关键技术。通过实战案例,帮助开发者搭建高性能、可扩展的 Go 微服务系统。

1

2026.02.10

热门下载

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

精品课程

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

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.1万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.5万人学习

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

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