python的csv.reader_如何使用Python的CSV DictReader解析CSV输出？

我已经集成了Salesforce的Bulk API,用于在我的Python项目中获取记录.使用“ Content-Type”：“ text / csv; charset = UTF-8’标头,它返回CSV输出.

"Id","Name","CreatedById","Salary","Base_Salary","Type","Pay_cycle","Description","Code"

"a0u90000003R4Y9AAK","Freelance Hadoop developer","005900000039GMdAAM","","","","","",""

"a0u90000003R5UPAA0","Senior Hadoop developer","005900000039GMdAAM","","","","","",""

"a0u90000003R5V3AAK","Freelance Webmaster","005900000039GMdAAM","","","","","",""

"a0u90000004ZMUwAAO","Full-Stack Web Developer - PHP and Python","005900000039GMdAAM","","","","","",""

现在,无需将其存储在CSV文件中,我要将这些记录存储在我的Db中.

DictReader是获取字段值的非常好的直接方法,但是我想它仅适用于(CSV)文件.

import csv

with open('output.csv') as csvfile:

reader = csv.DictReader(csvfile)

for row in reader:

print(row['Id'], row['Name'])

我正在尝试某些东西,但我不喜欢它.看到

objects = []

for line in iter(csv_data.splitlines()):

# Split by separator

data = line.split(",")

# Remove double quotes around the field value

objects.append([i[1:-1] for i in data])

什么是最优雅的方式？

解决方法:

您可以将csv字符串包装到io.StringIO对象中.然后,这将与csv模块完美配合：

data_str = '''"Id","Name","CreatedById","Salary","Base_Salary","Type","Pay_cycle","Description","Code"

"a0u90000003R4Y9AAK","Freelance Hadoop developer","005900000039GMdAAM","","","","","",""

"a0u90000003R5UPAA0","Senior Hadoop developer","005900000039GMdAAM","","","","","",""

"a0u90000003R5V3AAK","Freelance Webmaster","005900000039GMdAAM","","","","","",""

"a0u90000004ZMUwAAO","Full-Stack Web Developer - PHP and Python","005900000039GMdAAM","","","","","",""'''

# in python 2 you need to decode the string before passing it to StringIO

data_io = io.StringIO(data_str.decode('utf-8'))

## this is not needed in python 3

# data_io = io.StringIO(data_str)

reader = csv.DictReader(data_io)

for row in reader:

print row['Id'], row['Name']

标签：csv,python