按照

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: