Чтение нескольких файлов CSV из s3 с помощью boto3 - PullRequest
0 голосов
/ 16 января 2019

Код ниже читает файл CSV из AWS s3, используя Pycham на моей локальной машине.

# Read CSV from s3

import os
import boto3
import pandas as pd
import sys

if sys.version_info[0] < 3:
    from StringIO import StringIO  # Python 2.x
else:
    from io import StringIO

aws_id = 'XXXXXXXXXXXXXXX'
aws_secret = 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX'

client = boto3.client('s3', aws_access_key_id=aws_id, aws_secret_access_key=aws_secret)

bucket_name = 'bucket-name'
object_key = 'folder-name/test.csv'

csv_obj = client.get_object(Bucket=bucket_name, Key=object_key)

body = csv_obj['Body']

csv_string = body.read().decode('utf-8')

df = pd.read_csv(StringIO(csv_string))

x = df.head()

print(x)

Я хотел бы иметь возможность читать несколько файлов CSV одинаковым образом. Почти все, что находится в папке.

Мои файлы находятся в следующем каталоге:

bucket-name/folder-name/year=2018/month=01/file_032342.csv
bucket-name/folder-name/year=2018/month=02/file_434423.csv
bucket-name/folder-name/year=2018/month=03/file_343254.csv
bucket-name/folder-name/year=2018/month=04/file_544353.csv
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...