在 CSV python 中添加列并枚举它

我的 CSV 看起来像

John,Bomb,Dawn
3,4,5
3,4,5
3,4,5

我想在前面添加 ID 列，如下所示：

ID,John,Bomb,Dawn
1,3,4,5
2,3,4,5
3,3,4,5

使用枚举函数，但我不知道如何。到目前为止，这是我的代码：

import csv

with open("testi.csv", 'rb') as input, open('temp.csv', 'wb') as output:
        reader = csv.reader(input, delimiter = ',')
        writer = csv.writer(output, delimiter = ',')

        all = []
        row = next(reader)
        row.append('ID')
        all.append(row)
        count = 0
        for row in reader:
                count += 1
                while count:
                        all.append(row)
                        row.append(enumerate(reader, 1))
                        break
        writer.writerows(all)

输出全部错误：

John,Bomb,Dawn,ID
3,4,5,<enumerate object at 0x7fb2a5728d70>
3,4,5,<enumerate object at 0x1764370>
3,4,5,<enumerate object at 0x17643c0>

所以 ID 出现在最后，而它应该出现在开头，而且它甚至没有执行 1、2、3。出现一些奇怪的错误。

我可以建议下面的代码来解决您的问题：

import csv

with open("testi.csv", 'rb') as input, open('temp.csv', 'wb') as output:
    reader = csv.reader(input, delimiter = ',')
    writer = csv.writer(output, delimiter = ',')

    all = []
    row = next(reader)
    row.insert(0, 'ID')
    all.append(row)
    for k, row in enumerate(reader):
        all.append([str(k+1)] + row)
    writer.writerows(all)

更紧凑的代码可以是：

all = [['ID'] + next(reader)] + [[str(k+1)] + row for k, row in enumerate(reader)]

UPDATE（一些解释）：

你有错enumerate功能理解。enumerate应该用在for循环并且当你迭代时enumerate函数结果你得到元组的序列，其中第一个项目是列表中项目的有序数，第二个是项目本身。

But enumerate函数返回对象（docs）所以当你尝试将其转换为字符串时，它会调用__repr__魔术方法并将枚举对象转换为<enumerate object at ...>.

另外一句话，enumerate有助于避免循环中的额外计数器，例如您的count += 1多变的。

另外你这里有一个非常奇怪的代码：

while count:
    all.append(row)
    row.append(enumerate(reader, 1))
    break

这部分代码永远不能执行多次。