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

Question

我正在尝试从 morningstar 下载 CSV 内容，然后解析其内容。如果我将 HTTP 内容直接注入(inject) 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']