将 CSV 直接下载到 Python CSV 解析器中

Question

我正在尝试从 Morningstar 下载 CSV 内容，然后解析其内容。如果我将 HTTP 内容直接注入 Python 的 CSV 解析器，则结果格式不正确。然而，如果我将 HTTP 内容保存到文件 (/tmp/tmp.csv)，然后将文件导入 python 的 csv 解析器，结果是正确的。换句话说，为什么会：

def finDownload(code,report):
    h = httplib2.Http('.cache')
    url = 'http://financials.morningstar.com/ajax/ReportProcess4CSV.html?t=' + code + '&region=AUS&culture=en_us&reportType='+ report + '&period=12&dataType=A&order=asc&columnYear=5&rounding=1&view=raw&productCode=usa&denominatorView=raw&number=1'
    headers, data = h.request(url)
    return data

balancesheet = csv.reader(finDownload('FGE','is'))
for row in balancesheet:
    print row

返回：

['F']
['o']
['r']
['g']
['e']
[' ']
['G']
['r']
['o']
['u']
     (etc...)

代替：

[Forge Group Limited (FGE) Income Statement']

?

Answer 1

问题是由于文件的迭代是逐行完成的，而字符串的迭代是逐字符完成的。

您想要StringIO/cStringIO（Python 2）或io.StringIO（Python 3，感谢 John Machin 指点我），因此可以将字符串视为类似文件的对象：

Python 2：

mystring = 'a,"b\nb",c\n1,2,3'
import cStringIO
csvio = cStringIO.StringIO(mystring)
mycsv = csv.reader(csvio)

Python 3：

mystring = 'a,"b\nb",c\n1,2,3'
import io
csvio = io.StringIO(mystring, newline="")
mycsv = csv.reader(csvio)

两者都将正确保留引用字段内的换行符：

>>> for row in mycsv: print(row)
...
['a', 'b\nb', 'c']
['1', '2', '3']