從 S3 存儲桶中的 CSV 文件中讀取數據並將其存儲在 python 的字典中

Question

我正在嘗試從 S3 存儲桶中讀取 csv 文件並將其內容存儲到字典中。 樣本 csv 文件數據。 我想使用我的第一行作為鍵和后續行作為值。

name,origin,dest
xxx,uk,france
yyyy,norway,finland
zzzz,denmark,canada

我正在使用下面的代碼，它將整行存儲在字典中。 但是我想遍歷每一行並將每個字段存儲在一行中作為鍵值對。

{'name':'xxx','origin':'uk','dest':'france'}

s3 = boto3.client('s3')
obj = s3.get_object(Bucket = 'bucket_name', Key = 'logs/log.csv')
lines = obj['Body'].read().decode("utf-8").replace("'", '"')
lines = lines.splitlines()
if (isinstance(lines, str)):
        lines = (lines)

docData = {}
for line in lines:
        docData['content'] = str(line)

print(docData)

Answer 1

假如說

s3 = boto3.client('s3')
obj = s3.get_object(Bucket = 'bucket_name', Key = 'logs/log.csv')
lines = obj['Body'].read().decode("utf-8")

將 csv 的內容生成為字符串，然后您可以使用標准庫的csv模塊來獲取字典列表。

import csv
import io

buf = io.StringIO(lines)
reader = csv.DictReader(buf)
rows = list(reader)