Pandas の to_csv の保存先を s3 にする
January 13, 2022•36 words
SageMaker の Notebook から DataFrame を CSV にして S3 に保存する。
import pandas as pd
import boto3
from sagemaker import get_execution_role
role = get_execution_role()
bucket = 'BUCKET_NAME'
df = pd.DataFrame([1,2,3,4])
data_location = 's3://{}/results/test.csv'.format(bucket)
df.to_csv(data_location, index=False)