用标题写一个额外的列

时间:2015-04-17 11:23:09

标签: python csv

目前我有这个:

2个变量: varHeader:ExtraColumn varContent:ToBeFilledIn

        writer.writeheader()
        for line in reader:
            writer.writerow(line)

一切正常。问题是我有2个变量,varHeader和varContent。我失败了:

  1. 在writer.writeheader之前,我需要添加一个额外的列作为标题ExtraColumn,这是varHeader变量的内容。
  2. 每条线'需要将varContent的值添加到新创建的列
  3. 虽然这对我来说似乎很简单,但它让我大吃一惊,我无法完成它。猜猜这是一种典型的星期五疾病; - )

    感谢队友,

    来自新手。

1 个答案:

答案 0 :(得分:2)

这是你想要的吗?

from csv import DictReader, DictWriter

with open ...:
    reader = DictReader(in_f)
    writer = DictWriter(out_f, ['ExtraColumn'] + reader.fieldnames)
    writer.writeheader()
    for line in reader:
        line['ExtraColumn'] = varContent
        writer.writerow(line)