在 Python 中解析内存中的 CSV 数据

Question

当数据不在文件中时，有没有办法在Python中解析CSV数据？我将 CSV 数据存储在数据库中，并且想解析它。我正在寻找类似于 Ruby 的 CSV.parse 的东西。我知道 Python 有一个 CSV 类，但我在文档中看到的所有内容似乎都是处理文件而不是内存中的 CSV 数据。

（并且在数据进入数据库之前不能解析数据。）

（请不要告诉我不要将 CSV 数据存储在数据库中。就数据库而言，我知道我在做什么。 ）

Answer 1

关于 python csv 模块的文件没有特殊区别。您可以使用 StringIO 将字符串包装为类似文件的对象。

Answer 2

这就是为什么你应该使用cStringIO.StringIO（Python 3.x中的io.StringIO）而不是一些DIY组装：

>>> import csv
>>> from cStringIO import StringIO
>>> fromDB = '"Column\nheading1",hdng2\r\n"data1\rfoo","data2\r\nfoo"\r\n'
>>> sources = [StringIO(fromDB), fromDB.splitlines(True),
...     fromDB.splitlines(), fromDB.split("\n")]
>>> for i, source in enumerate(sources):
...     print i, list(csv.reader(source))
...
0 [['Column\nheading1', 'hdng2'], ['data1\rfoo', 'data2\r\nfoo']] # OK
1 [['Column\nheading1', 'hdng2'], ['data1\rfoo', 'data2\r\nfoo']] # OK
2 [['Columnheading1', 'hdng2'], ['data1foo', 'data2foo']]         # 3 errors
3 [['Columnheading1', 'hdng2'], ['data1\rfoo', 'data2\rfoo'], []] # 3 errors
>>>

使用guff.splitlines(True)<不推荐使用 /code>，因为它比 StringIO(guff) 更有可能导致任何阅读你代码的人都不知道它的作用。

Answer 3

http://docs.python.org/library/csv.html

csv.reader(csvfile)

csvfile 可以是任何对象
支持迭代器协议和
每次 next() 都返回一个字符串
方法被调用 — 文件对象和
列表对象都适合。

例如，如果您有字符串中的数据库内容，您可以像这样解析它

import csv

fromDB = "1,2,3\n4,5,6"

reader = csv.reader(fromDB.split("\n"))
for row in reader:
  print("New row")
  for col in row:
    print("  ", col)

Answer 4

使用 stringio 模块，它允许您将字符串装饰为类似文件的对象。这样您就可以将 stringio “文件”传递给 CSV 模块进行解析（或您可能使用的任何其他解析器）。