讲师中心 微信公众号
AI工具推荐 视频效率加速

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

夜涛姑娘_7514

夜涛姑娘_7514

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

|

363人浏览过

|

来源于php中文网

原创

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

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

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

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

Knowledge Graph - Csv Graph Loader Generator
Knowledge Graph - Csv Graph Loader Generator

从 CSV 数据集生成图数据库加载器和三元组映射。将表格 CSV 数据转换为图数据库可直接使用的节点、边和三元组。

下载

✅ 推荐方案:使用 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工具

更多
立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

豆包大模型

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

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

DeepSeek

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

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

WorkBuddy

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

VibeKnow
VibeKnow Hot

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

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

相关专题

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

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

340

2025.12.04

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

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

252

2026.01.31

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

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

691

2026.04.13

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

20

2026.10.28

Vite 如何搭配 Vue.js
Vite 如何搭配 Vue.js

Vite搭配Vue.js专题汇集创建Vue 3项目、配置开发环境、路径别名、Vue插件、代理、多环境构建与调试的已审核文章,帮助读者完成从初始化到日常开发的配置。

0

2026.10.28

Vite 项目目录怎么看
Vite 项目目录怎么看

Vite项目目录专题汇集入口文件、src源码、public静态资源、路径别名、项目根目录和构建输出目录相关的已审核文章,帮助读者理解各文件夹的用途与引用方式。

0

2026.10.28

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

120

2026.10.22

Gradle Wrapper配置教程
Gradle Wrapper配置教程

本专题介绍Gradle Wrapper的基本作用,涵盖文件组成、生成方法、版本配置和任务调用,帮助你理解项目如何使用指定的Gradle版本,减少不同开发环境带来的构建差异。

60

2026.10.22

Golangci-lint检查器使用教程
Golangci-lint检查器使用教程

本专题整理Golangci-lint常用Linters的用途与配置方法,覆盖错误处理、无效代码、安全、复杂度、命名和依赖检查,帮助项目合理选规则。

60

2026.10.22

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

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

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