如何高效合并多个小型CSV文件(避免列表累积导致的性能瓶颈)

大杰酱_4695

大杰酱_4695

2026-10-16

143人浏览

原创

如何高效合并多个小型CSV文件(避免列表累积导致的性能瓶颈)

本文介绍针对小规模CSV文件(如6列×5行)批量合并的优化策略,重点规避list.append()累积DataFrame带来的内存与时间开销,推荐使用生成器式读取+单次pd.concat()、分治合并或流式写入等高性能方案。

本文介绍针对小规模csv文件(如6列×5行)批量合并的优化策略,重点规避`list.append()`累积dataframe带来的内存与时间开销,推荐使用生成器式读取+单次`pd.concat()`、分治合并或流式写入等高性能方案。

在处理大量小型CSV文件(例如每文件仅6列×5行,含1行表头)时,采用“逐个读取 → append() 到列表 → 最后pd.concat()”的模式虽直观,却存在显著性能缺陷:

  • 每次list.append()本身开销虽小,但当文件数达数百或上千时,Python列表动态扩容引发多次内存复制;
  • 更关键的是,pd.DataFrame对象在Python层是重量级对象,频繁创建和引用会加剧GC压力与内存碎片;
  • pd.concat()对长列表执行时,内部需遍历全部DataFrame并校验schema一致性,效率随列表长度非线性下降。

✅ 推荐优化方案(按优先级排序):

1. 使用生成器 + 单次pd.concat()(最简高效)
避免中间列表,直接将read_csv调用惰性封装为生成器,交由pd.concat()统一处理:

import pandas as pd
import os
from pathlib import Path

def read_csv_generator(file_paths, **read_csv_kwargs):
    for fp in file_paths:
        yield pd.read_csv(fp, **read_csv_kwargs)

# 示例:每8个文件一组合并
all_files = []
for root, _, files in os.walk(base_dir):
    for file in files:
        if file.endswith('.csv'):
            all_files.append(os.path.join(root, file))

# 分组处理(每8个一组)
for i in range(0, len(all_files), 8):
    batch = all_files[i:i+8]
    # 关键:不构建列表,直接传生成器
    combined_df = pd.concat(
        read_csv_generator(batch, sep='\t'),
        ignore_index=True,
        sort=False
    )
    combined_df.to_csv(f"batch_{i//8 + 1}.csv", index=False, sep='\t')

✅ 优势:零中间列表、内存占用恒定(仅当前DataFrame)、代码简洁、充分利用pd.concat的底层优化。

QuickBooks Mileage CSV
QuickBooks Mileage CSV

使用 qbo-mileage CLI 及用户凭证,从 Airtable、Outlook 或 Google Calendar 记录生成 QuickBooks Online 里程 CSV 文件。

下载

2. 分治式两两合并(适合超大文件集或内存受限场景)
当文件总数极大(如>10,000)且单次concat仍显吃力时,可模拟归并排序逻辑,递归两两合并临时文件:

def merge_two_csv(path1, path2, output_path, **kwargs):
    df1 = pd.read_csv(path1, **kwargs)
    df2 = pd.read_csv(path2, **kwargs)
    pd.concat([df1, df2], ignore_index=True).to_csv(output_path, index=False, **kwargs)

def merge_files_in_batches(file_list, batch_size=8, temp_prefix="temp_"):
    # 第一阶段:每batch_size个文件合并为一个临时文件
    temp_files = []
    for i in range(0, len(file_list), batch_size):
        batch = file_list[i:i+batch_size]
        temp_out = f"{temp_prefix}{i}.csv"
        pd.concat(
            (pd.read_csv(f, sep='\t') for f in batch),
            ignore_index=True
        ).to_csv(temp_out, index=False, sep='\t')
        temp_files.append(temp_out)

    # 第二阶段:两两合并临时文件(可迭代进行,无需递归函数)
    while len(temp_files) > 1:
        new_temp_files = []
        for i in range(0, len(temp_files), 2):
            if i + 1 <p><strong>3. 流式写入(零内存DataFrame,极致轻量)</strong><br>
若后续分析不强依赖Pandas(如仅需统计或导出),可跳过DataFrame,直接用<code>csv</code>模块拼接内容(跳过重复表头):</p><pre class="brush:php;toolbar:false;">import csv

def fast_concat_csv_files(file_list, output_path, sep='\t'):
    with open(output_path, 'w', newline='') as out_f:
        writer = csv.writer(out_f, delimiter=sep)
        for i, fp in enumerate(file_list):
            with open(fp, 'r') as f:
                reader = csv.reader(f, delimiter=sep)
                rows = list(reader)
                if i == 0:
                    # 写入首文件的表头
                    writer.writerows(rows)
                else:
                    # 跳过后续文件的表头(rows[0]),只写数据行
                    writer.writerows(rows[1:])

# 调用示例
fast_concat_csv_files(batch, "output.csv")

⚠️ 注意事项:

  • 所有方案中,确保各CSV文件列名、列顺序、数据类型严格一致,否则pd.concat可能隐式转换或报错;建议在读取时显式指定dtype和usecols;
  • 若文件编码不统一,务必在read_csv中添加encoding='utf-8'(或自动检测);
  • 对于您描述的“每模型8个输出”的业务逻辑,建议将batch_size=8与模型ID绑定(如按文件名前缀分组),避免跨模型混拼。

综上,摒弃list.append()累积DataFrame是性能跃升的关键第一步。对典型的小型CSV场景,方案1(生成器+单次concat)在简洁性、可维护性与性能间取得最佳平衡,应作为首选实践。

相关文章

数码产品性能查询
数码产品性能查询

该软件包括了市面上所有手机CPU,手机跑分情况,电脑CPU,电脑产品信息等等,方便需要大家查阅数码产品最新情况,了解产品特性,能够进行对比选择最具性价比的商品。

下载

相关标签:

csv 性能瓶颈 csv文件

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

相关专题

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

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

2025.12.04

320

15

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

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

2026.01.31

252

21

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

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

2026.04.13

511

25

Revel框架模板渲染与表单验证
Revel框架模板渲染与表单验证

本专题围绕Revel框架视图层开发,介绍模板语法、变量输出、布局复用、表单提交、参数绑定与验证规则,帮助开发者完成安全规范的页面交互。

2026.10.16

0

15

Revel框架MVC路由与控制器
Revel框架MVC路由与控制器

本专题聚焦Revel框架MVC核心机制,讲解路由定义、控制器Action、参数绑定、请求响应和反向路由,并结合常见场景梳理接口请求处理流程。

2026.10.16

0

15

Revel框架入门与安装配置
Revel框架入门与安装配置

本专题从Revel框架环境搭建讲起,系统介绍命令行工具安装、项目创建、目录结构、配置文件及热代码重载,帮助开发者快速完成首个Go Web应用。

2026.10.16

0

15

Linux 新手先学哪些命令
Linux 新手先学哪些命令

Linux新手命令专题汇集终端基础、文件目录操作、文本查看、软件安装和权限管理等入门文章,帮助按实际任务练习常用命令并建立命令行操作习惯。

2026.10.15

0

20

MySQL 忘记密码怎么办
MySQL 忘记密码怎么办

MySQL忘记密码专题汇集不同版本与系统下的root密码重置方法,覆盖MySQL 5.7、8.0、Mac、Linux及本地集成环境,并提示恢复权限与重新登录的检查步骤。

2026.10.15

0

18

Vue.js 中 ref 和 reactive 的区别
Vue.js 中 ref 和 reactive 的区别

Vue.js中ref和reactive的区别专题汇集两种响应式API的定义、访问方式、对象解构、侦听与性能取舍等文章,帮助结合实际状态类型选择合适写法。

2026.10.15

0

20

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Swoole手册
Swoole手册

共0课时 | 0人学习