在python中用head列出csv

Question

我写了一个脚本，它给出了如下列表作为输出。

['red', '361', '0']
['blue', '1', '0']
['orange', '77', '0']
['cream', '660', '73']
['ivory', '159', '0']

这个列表非常大，我想把输出内容写到一个csv中，上面有标题，如下所示。

color | total | fail

这就是我努力实现它的方式

with open('colour.csv', 'wb') as csvfile:
    header = ['color', 'total', 'fail']
    writer = csv.writer(csvfile, delimiter=',')
    writer.writerow(header)
    for row in output:
        writer.writerow(output)

但我得到的结果并不像预期的那样。生成csv，并将列表的最后一行打印三次到csv。我找不到为什么不打印所有其他行或为什么最后一行打印三次？ 下面是我得到的示例输出：

color | total | fail
ivory | 159 | 0
ivory | 159 | 0
ivory | 159 | 0

这就是我期望的结果：

color | total | fail
red | 361 | 0 
blue | 1 | 0
orange | 77 | 0
cream | 660 | 73
ivory | 159 | 0

有人能帮我做这个吗？

Answer 1

代码-

import csv

arr = [['red', '361', '0'],
      ['blue', '1', '0'],
      ['orange', '77', '0'],
      ['cream', '660', '73'],
      ['ivory', '159', '0']]

with open('output.csv','w') as f:
    writer = csv.writer(f)
    writer.writerow(['color', 'total', 'fail'])
    writer.writerows(arr)