python - 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)