pandas.read_parquet()读取2G的parquet文件报错:pyarrow.lib.ArrowMemoryError: malloc of size 5371271360 failed

本文介绍了如何使用PyArrow底层接口逐块读取和处理Parquet文件,以及利用Dask的并行计算能力和懒加载特性高效地读取大文件。这两种方法都适用于大规模数据处理场景。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

解决方式:

1:使用 PyArrow 底层接口:
PyArrow 是 Pandas 在背后使用的库来处理 Parquet 文件。你可以直接使用 PyArrow 来逐块读取数据:

import pyarrow.parquet as pq

parquet_file = pq.ParquetFile('big_file.parquet')

for i in range(parquet_file.num_row_groups):
    row_group = parquet_file.read_row_group(i)
    # 处理这个 row group 的数据,例如转换成 Pandas DataFrame
    df = row_group.to_pandas()
    # 现在你可以对 df 进行处理

2:使用 Dask:
Dask 是一个并行计算库,它有自己的 read_parquet 函数,它支持分块处理和懒加载,这意味着数据只有在需要计算时才会被加载到内存中:

import dask.dataframe as dd

ddf = dd.read_parquet('your_file.parquet', chunksize='100MB')
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值