csv - 将子进程的输出转换为 csv.reader 对象

Question

有没有办法从 subprocess 获取输出并将其转换为可迭代的 csv.reader 或 csv.DictReader 对象？这是我一直在尝试的代码:

p2 = subprocess.Popen("sort command...", stdout=subprocess.PIPE)
output = p2.communicate()[0]
edits = csv.reader(output, delimiter="\t")

基本上，我正在对一个大的 CSV 文件进行排序，然后我想将它作为 csv.reader 对象放入 Python 中。

我得到的错误是

Error: iterator should return strings, not int (did you open the file in text mode?)

有没有办法将此字节流视为 csv.reader 对象，还是我想错了？

Answer 1

这是 Python 3 中的一个问题。CSV 模块需要 unicode 输入，而不是字节字符串。除此之外，csv.reader()需要一个可迭代的对象，例如打开的文件或字符串列表。试试这个:

encoding = 'ascii'    # specify the encoding of the CSV data
p2 = subprocess.Popen(['sort', '/tmp/data.csv'], stdout=subprocess.PIPE)
output = p2.communicate()[0].decode(encoding)
edits = csv.reader(output.splitlines(), delimiter=",")
for row in edits:
    print(row)

如果 /tmp/data.csv包含(我使用逗号作为分隔符):

1,2,3,4
9,10,11,12
A B C D
5,6,7,8

那么输出将是:

['1', '2', '3', '4']
['5', '6', '7', '8']
['9', '10', '11', '12']
['A B C D']