python - 在 python 中将列表写入 csv 文件的末尾

Question

我有以下代码:

included_cols = [1, 3, 5]

w = csv.writer(open(args.output,'w'))
with open('input.csv', 'r') as input_file:
    fid = csv.reader(input_file, delimiter = ',')
    for row in fid:
        item = [row[i] for i in included_cols)]

        w.writerows(item)

对于每次迭代，row 类似于 [1, 2]...[3, 4] 等。但事实并非如此将每个列表作为一行写入 w...有什么想法可以解决这个问题吗？

Answer 1

如果您尝试仅从 csv 文件中选择某些列，则可以使用以下命令:

import csv
from itertools import imap
from operator import itemgetter

with open('input.csv', 'rb') as fin, open('output.csv', 'wb') as fout:
    csvin = csv.reader(fin)
    csvout = csv.writer(fout)
    csvout.writerows(imap(itemgetter(1, 3, 5), csvin))

您的主要问题是您使用的 .writerows 接受一个可迭代对象，并将每个迭代视为一行，因此给它 [1, 2, 3] 它将写出 3 行...而使用 .writerow 将期望一个可迭代对象，并写出一行，并将其作为列...