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

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'}