writerow()或writerows()循环

Question

我有以下代码:

import csv
input_file = csv.reader(open('template.csv','r+')) #template.csv has 10 lines
output = open('output.csv', 'wb')
wr = csv.writer(output, quoting=csv.QUOTE_ALL)

elem = [1,2,3,4,5]

for i in elem:
    wr.writerows(input_file)

由于文件'template.csv'有10行,我期望输出10 x 5行 - 但只出现10行.

如何嵌套writerow/rows循环？

Answer 1

你input_file是一个迭代器,一旦筋疲力尽就不会为你重新开始.将结果存储在列表中并写入:

input_file = csv.reader(open('template.csv','r+')) #template.csv has 10 lines
input_lines = list(input_file)