1 回答

TA貢獻1826條經驗 獲得超6個贊
您可以將流方法與boto / s3一起使用,但您必須定義自己的類文件對象 AFAIK。
幸運的是,有smart_open可以幫你處理這個問題;它還支持GCS、Azure、HDFS、SFTP等。以下是使用大量銷售數據樣本
的 示例:
import boto3
from smart_open import open
session = boto3.Session()? # you need to set auth credentials here if you don't have them set in your environment
chunk_size = 1024 * 1024? # 1 MB
f_in = open("s3://mybucket/2m_sales_records.csv.gz", transport_params=dict(session=session), encoding="utf-8")
f_out = open("s3://mybucket/2m_sales_records.csv", "w", transport_params=dict(session=session))
byte_count = 0
while True:
? ? data = f_in.read(chunk_size)
? ? if not data:
? ? ? ? break
? ? f_out.write(data)
? ? byte_count += len(data)
? ? print(f"wrote {byte_count} bytes so far")
f_in.close()
f_out.close()
示例文件有200 萬行,壓縮后為75 MB,未壓縮為238 MB。
我將壓縮文件上傳到mybucket并運行下載該文件的代碼,提取內存中的內容并將未壓縮的數據上傳回 S3。
在我的計算機上,該過程大約需要78 秒(高度依賴于互聯網連接速度),并且從未使用超過95 MB的內存;我認為如果需要的話,您可以通過覆蓋smart_open中 S3 分段上傳的部分大小來降低內存要求。
DEFAULT_MIN_PART_SIZE = 50 * 1024**2
"""Default minimum part size for S3 multipart uploads"""
MIN_MIN_PART_SIZE = 5 * 1024 ** 2
"""The absolute minimum permitted by Amazon."""
添加回答
舉報