Python与Parquet的组合是当前大数据分析中最优的列式存储方案之一,在压缩率、查询性能和生态兼容性上均优于传统CSV或JSON格式。 在2026年,数据量持续增长,Python开发者需要一种高效的数据存储格式来应对内存和I/O瓶颈,Parquet作为Apache Hadoop生态的核心列式存储格式,天然支持嵌套数据、谓词下推和压缩优化,而Python社区提供了pandas、PyArrow、fastparquet等多个库来无缝对接,我们从实操角度,全面掌握Python处理Parquet的完整流程。
为什么选择Python处理Parquet
Parquet的列式存储设计让它在分析场景中具备天然优势,传统行式存储(如CSV)读取整行数据,即使只需少数列,也得加载全部字段,造成I/O浪费,Parquet按列存储,查询时只需读取目标列,大幅减少磁盘读取量,据统计,在典型大数据场景下,Parquet可减少60%以上的存储空间,同时将查询速度提升数倍。
Python生态对Parquet的支持非常成熟,pandas从0.21版本开始内置read_parquet和to_parquet方法,默认依赖PyArrow或fastparquet引擎,PyArrow作为Apache Arrow的Python绑定,提供零拷贝读取和列式计算能力,是官方推荐的高性能引擎,fastparquet轻量高效,适合纯Python环境,行业共识认为,基于PyArrow的Parquet读写是Python大数据处理的最佳实践。
python parquet 读取与写入全流程详解
安装必要的库
推荐使用conda或pip安装:
- 核心库:
pandas、pyarrow、fastparquet - 可选:
numpy、boto3(用于S3)、s3fs(分布式文件系统)
pip install pandas pyarrow fastparquet s3fs
使用Pandas读写Parquet
Pandas的read_parquet和to_parquet是最直接的接口,默认引擎为PyArrow,也可指定fastparquet。
写入操作:
import pandas as pd
df = pd.read_csv('data.csv')
# 写入单文件
df.to_parquet('data.parquet', engine='pyarrow', compression='snappy')
# 写入分区数据集(按年/月分区)
df.to_parquet('data_partitioned', partition_cols=['year', 'month'])
读取操作:
# 读取单文件
df = pd.read_parquet('data.parquet', engine='pyarrow')
# 读取分区数据集,自动合并分区
df = pd.read_parquet('data_partitioned', engine='pyarrow')
# 只读取部分列,减少内存
df = pd.read_parquet('data.parquet', columns=['id', 'value'])
使用PyArrow原生API
PyArrow的底层API提供更细粒度的控制,适合处理超大文件或流式场景。
写入:
import pyarrow.parquet as pq import pyarrow as pa table = pa.Table.from_pandas(df) pq.write_table(table, 'data.parquet', compression='zstd', row_group_size=65536)
读取:
table = pq.read_table('data.parquet', columns=['id', 'value'])
df = table.to_pandas()
# 分块读取,避免内存爆满
batches = pq.read_table('data.parquet', use_pandas_metadata=True).to_batches()
for batch in batches:
df_batch = batch.to_pandas()
process(df_batch)
处理分区数据集
Parquet分区是Hive风格的目录结构,如data/year=2026/month=01/file.parquet,pandas的read_parquet支持直接读取分区目录,并自动将分区字段作为列返回,pyarrow的ParquetDataset提供更灵活的分区过滤。
import pyarrow.parquet as pq
dataset = pq.ParquetDataset('data_partitioned', filters=[('year', '=', 2026)])
table = dataset.read()
python parquet 性能对比:列式存储 vs 行式存储
读写速度与存储空间对比
在相同数据集(100万行,10列,含整数和浮点数)下,典型测试结果如下:
| 格式 | 存储空间 | 读取时间(全部列) | 读取时间(2列) | 写入时间 |
|---|---|---|---|---|
| CSV | 100 MB | 1 s | 9 s | 8 s |
| JSON | 120 MB | 5 s | 2 s | 5 s |
| Parquet(snappy) | 18 MB | 4 s | 08 s | 6 s |
| Parquet(zstd) | 14 MB | 5 s | 09 s | 9 s |
(数据来源:基于社区公开基准测试,实际结果因硬件和配置而异)
Parquet在读取速度和存储空间上优势明显,尤其是只读取部分列时,速度相差近20倍,写入速度略慢于CSV,但考虑到后续查询性能,这个代价完全值得。
压缩算法的选择
Parquet支持多种压缩算法,需根据场景权衡:
- snappy:速度最快,压缩比一般,适合常规场景。
- zstd:压缩比高,速度接近snappy,推荐用于归档或网络传输。
- gzip:压缩比最高,但压缩解压慢,适合存储长期不用的数据。
- lz4:解压速度极快,但压缩比低,适合实时流处理。
PyArrow默认使用snappy,但可通过compression='zstd'切换,测试表明,zstd可将文件再缩小20%-30%,解析时间仅增加10%,是性价比之选。
进阶技巧:python parquet 优化与常见错误
分区与排序
Parquet的谓词下推功能依赖于分区和排序键,合理设计分区字段(如日期、地区)能大幅减少扫描数据量,排序键(sorting columns)则让相邻记录在磁盘上连续,进一步提升压缩率和过滤效率。
# 使用PyArrow写入时指定排序键
pq.write_table(table, 'sorted.parquet', row_group_size=65536,
sort_by='value')
内存不足的解决方案
处理超大规模Parquet文件时,一次性读取可能导致OOM,推荐以下策略:
- 分块读取:使用
pq.read_table的use_threads=False并配合to_batches迭代。 - 列过滤:只读取需要的列,减少内存占用。
- 使用dask或polars:这些库内置延迟计算和分块处理,适合超大数据集。
# Dask读取Parquet示例
import dask.dataframe as dd
ddf = dd.read_parquet('large_dataset.parquet', columns=['id', 'name'])
# 触发计算
result = ddf.compute()
模式演变与兼容性
Parquet支持模式演变(schema evolution),允许新增或删除列而不影响现有数据,pandas的to_parquet默认使用PyArrow的Strict类型,如果DataFrame列类型与之前不一致,可能报错,建议使用schema参数强制统一。
# 读取时指定schema,类型不匹配时自动转换
import pyarrow.types as pt
schema = pa.schema([('id', pa.int64()), ('name', pa.string())])
table = pq.read_table('data.parquet', schema=schema)
Python与Parquet的未来在数据工程
掌握Parquet就是掌握大数据处理的效率密码。 从单机脚本到分布式计算,Parquet的列式存储、压缩能力和生态兼容性让数据存储不再成为瓶颈,无论是日常分析报表,还是构建数据湖,Python+Parquet的组合都是值得信赖的基础设施。
python parquet 常见问题解答
Q1: python parquet 怎么安装?
A1: 推荐使用pip install pandas pyarrow fastparquet,PyArrow是性能最优的引擎,fastparquet适合纯Python环境,如果需要S3支持,安装s3fs和boto3。
Q2: python parquet 读取时内存不足怎么办?
A2: 采用分块读取策略:使用PyArrow的read_table配合to_batches迭代,或者用pandas.read_parquet的chunksize参数(需指定引擎为pyarrow),对于超大数据集,建议使用Dask或Polars,它们支持延迟计算和溢出存储。
Q3: python parquet 与CSV如何互相转换?
A3: 使用pandas的read_csv和to_parquet,或PyArrow的read_csv和write_table,转换时注意列类型自动推断,CSV中的日期列需手动解析,推荐先用pd.read_csv读取,再to_parquet,最后用read_parquet验证,确保数据完整性。
首发原创文章,作者:世雄 - 原生数据库架构专家,如若转载,请注明出处:https://idctop.com/article/502040.html



