Question

我有一些以逗号分隔的输入，我想解析成字典，所以csv.DictReader似乎很合适。但是，输入已经是字符串形式，而不是作为csv.DictReader想要的接口的文件。

有没有办法直接使用csv.DictReader字符串？

Answer 1

您可以使用StringIO（或io.BytesIO / io.StringIO）：

>>> import StringIO
>>> import csv
>>>
>>> f = StringIO.StringIO(u'''field1,field2,field3
... 1,2,3
... 4,5,6
... 7,8,9
... ''')
>>>
>>> for row in csv.DictReader(f):
...     print row
...
{'field2': '2', 'field3': '3', 'field1': '1'}
{'field2': '5', 'field3': '6', 'field1': '4'}
{'field2': '8', 'field3': '9', 'field1': '7'}

Answer 2

您可以使用任何可迭代的生产线，它不仅仅是一个文件对象。您可以将字符串拆分为包含str.splitlines()的行列表：

for row in csv.DictReader(inputtext.splitlines()):
    print row

演示：

>>> import csv
>>> demostring = '''\
... foo,bar,baz
... 42,Monty,Python
... 38,Ham,Spam
... '''
>>> for row in csv.DictReader(demostring.splitlines()):
...     print row
... 
{'foo': '42', 'baz': 'Python', 'bar': 'Monty'}
{'foo': '38', 'baz': 'Spam', 'bar': 'Ham'}

你可以在没有文件的情况下使用csv.DictReader吗？

2 个答案: