将 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']