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

如何使用 Polars 从 AWS S3 高效读取 Parquet 文件

酷晨同学_3084

酷晨同学_3084

发布时间:2026-03-31 10:16:10

|

405人浏览过

|

来源于php中文网

原创

本文详解在 Polars 中通过 storage_options 正确配置 AWS 凭据(含 Session Token)以安全、可靠地读取 S3 上的 Parquet 文件,避免 403 Forbidden 等权限错误。

本文详解在 polars 中通过 `storage_options` 正确配置 aws 凭据(含 session token)以安全、可靠地读取 s3 上的 parquet 文件,避免 403 forbidden 等权限错误。

在使用 Polars 读取 AWS S3 上的 Parquet 文件时,仅传递 aws_access_key_id 和 aws_secret_access_key 往往不够——尤其当你的 AWS Profile 使用了基于角色的临时凭证(如通过 aws sts assume-role 或 SSO 登录获取)时,必须显式传入 aws_session_token,否则将触发 ComputeError: Client error with status 403 Forbidden。这是因为 Polars 的底层对象存储客户端(object_store)不自动解析本地 ~/.aws/credentials 或 ~/.aws/config 中的 profile 配置,也无法隐式加载 session token。

✅ 推荐做法:使用 boto3 获取完整冻结凭证

以下代码展示了最健壮、跨环境兼容的写法,适用于 IAM 用户密钥、角色扮演(AssumeRole)、SSO 登录等多种认证场景:

import polars as pl
import boto3

# 指定本地 AWS Profile 名称(需已通过 aws-cli 配置)
profile_name = "my-prod-profile"
s3_path = "s3://my-s3-bucket/data/output.parquet"

# 创建 boto3 session 并提取完整凭证(含 token)
session = boto3.Session(profile_name=profile_name)
credentials = session.get_credentials().get_frozen_credentials()

# 构建 storage_options —— 必须包含三项核心凭证 + region
storage_options = {
    "aws_access_key_id": credentials.access_key,
    "aws_secret_access_key": credentials.secret_key,
    "aws_session_token": credentials.token,  # ⚠️ 关键!不可省略
    "aws_region": session.region_name or "us-east-1",  # 建议显式指定
}

# 使用 read_parquet(立即执行)或 scan_parquet(惰性执行)
df = pl.read_parquet(s3_path, storage_options=storage_options)
# 或:lf = pl.scan_parquet(s3_path, storage_options=storage_options).collect()

⚠️ 注意事项与最佳实践

  • 不要硬编码凭证:始终优先使用 boto3.Session(profile_name=...) 动态获取凭证,而非手动设置环境变量或明文密钥。
  • aws_session_token 是必需的:若 credentials.token 为 None(例如使用长期 IAM 密钥),可安全省略该字段;但只要 profile 启用了 MFA、SSO 或 AssumeRole,token 就一定存在且必须传递。
  • Region 必须匹配:aws_region 应与 S3 bucket 所在区域一致,否则可能引发签名错误(SignatureDoesNotMatch)。
  • 惰性 vs 立即加载:
    • pl.read_parquet():直接返回 DataFrame,适合小到中等规模数据;
    • pl.scan_parquet():返回 LazyFrame,支持链式过滤/投影下推(pushdown),大幅提升大文件处理效率,推荐用于生产 ETL 流程。
  • 替代方案(高级用户):若希望完全复用 AWS CLI 配置(包括 credential_process 或 sso),可改用 polars 0.20.20+ 版本支持的 anonymous=False + 自动凭据链(需确保 AWS_PROFILE 环境变量已设,且 boto3 可访问):
    import os
    os.environ["AWS_PROFILE"] = "my-prod-profile"  # 仅当未显式传 storage_options 时生效
    df = pl.read_parquet("s3://bucket/key.parquet")  # Polars 会尝试自动加载

    但该方式透明性低、调试困难,生产环境仍强烈推荐显式传参。

✅ 验证是否成功

运行后可通过以下方式快速验证连接与权限:

print(df.schema)      # 查看列结构
print(df.head(3))     # 预览前 3 行
print(df.estimated_size())  # 查看内存估算(确认已实际加载)

若仍报错,请检查:

  • S3 路径是否拼写正确(注意 s3://bucket-name/prefix/file.parquet 格式);
  • IAM Policy 是否授予 s3:GetObject 权限(及 s3:ListBucket,若需自动解析目录);
  • Bucket 是否启用了 Block Public Access?是否位于 VPC Endpoint 内而缺少相应策略?

掌握这一模式后,你不仅能稳定读取 Parquet,还可无缝迁移到 Delta Lake、NDJSON、CSV 等其他 S3 支持格式——只需替换 read_parquet 为对应函数即可。

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

热门AI工具

更多
切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

豆包大模型

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

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

WorkBuddy

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

DeepSeek

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

超级简历WonderCV

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

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1671

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4224

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1669

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

24477

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2987

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3027

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1163

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

596

2023.08.10

FrankenPHP集成Laravel详细教程
FrankenPHP集成Laravel详细教程

本专题提供FrankenPHP集成Laravel的详细配置指南,全面解析运行原理、开发环境搭建、Caddyfile配置、Octane工作模式、数据库连接、队列任务、定时任务和生产环境优化,解决部署过程中常见的报错与兼容性问题。

40

2026.10.08

热门下载

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

精品课程

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

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