python - Python CSV .. 将 CSV 文件输出到具有不同列数的另一个 CSV 文件

Question

我有一个海量 CSV 文件，我想对几个字段进行一些计算并将结果输出到另一个 CSV 文件。

假设我的file1.csv. 这是我的示例代码：

import csv
file1 = csv.reader(open('file1.csv', 'rb'), delimiter=';') #traffic

for record in file1:
    print record[0], int(record[1]) * int(record[4])

现在..我想将这些行保存在一个新的 csv 文件中。但我被困在那里。 writerrow()方法只接受整行，而不是像我放在for循环中的模式。

有什么建议吗？？

Answer 1

writerow需要一个可迭代的。您可以通过创建列表轻松组成新行

# new_csv_writer = open file for writing
for record in file1:
    new_csv_writer.writerow([record[0], int(record[1]) * int(record[4])])

上面将写一行 2 列