按照

Question

在csv文件中填充一行这里是我想用python更新的一个csv文件。 它有3列：

col1 = [1,2,3,4,5,6,7,8,9,10] 
col2 = [A, A, A, B, C, C, C, C, A, A] 
col3 = [] 

对于COL3每个细胞，我想填补一个ID号，将改变的COL2变化的内容，只有当。

在我的例子，我会得到col3 = [1,1,1,2,3,3,3,3,4,4]

任何人给我的方法是什么？

Answer 1

在Python 3：

from itertools import groupby 
from operator import itemgetter 
import csv 

with open('f1.csv', newline='') as f_in, open('f2.csv', 'w', newline='') as f_out: 
    reader = csv.reader(f_in) 
    writer = csv.writer(f_out) 
    for i, (key, group) in enumerate(groupby(reader, key=itemgetter(1)), start=1): 
     for row in group: 
      writer.writerow(row + [i]) 

如果使用Python 2.7，使用的语句改为

with open('f1.csv', 'rb') as f_in, open('f2.csv', 'wb') as f_out: