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

Question

我有一个大量的CSV文件，我想在几个字段上做一些计算，然后将结果输出到另一个CSV文件。

让我们假设我file1.csv上有12个字段。 这是我的示例代码：

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列