Python正則表達式替換爲CSV

Question

字符串我有格式化爲以下csv文件：

cat, mammal[1] 
    shark, fish[2] 
    dog, mammal[3] 
    tiger, mammal[4] 
    salmon, fish[5] 

我想換成含有哺乳動物的所有行與方括號一起。

輸出應該如下：

cat, mam 
shark, fish[2] 
dog, mam 
tiger, mam 
salmon, fish[5] 

到目前爲止，我有一個代碼，讀/寫csv文件：

import csv 


with open('animals.csv', 'r') as fin, open("out.csv",'w') as fout: 
     writer = csv.writer(fout) 
     for row in csv.reader(fin): 
      re.sub(???) #stuck at writing the regular expression 
      writer.writerow(row) 

Answer 1

您可以使用下面的正則表達式的替換：

for row in csv.reader(fin): 
    row[1] = re.sub(r'(\s*mam)mal\[\d+\]', '\1', row[1]) 
    writer.writerow(row) 

參見demonstration。

Answer 2

無需正則表達式在這裏：

for row in csv.reader(fin): 
    if row[1].startswith("mammal["): 
     row[1] = "mam" 

性能方面是最好的，因爲

• 沒有正則表達式
• 字符串替換隻有在比賽中，保持不變否則

甚至更快的發電機理解和writerows：

with open('animals.csv', 'r') as fin, open("out.csv",'w') as fout: 
    csv.writer(fout).writerows([row[0],"mam"] if row[1].starswith("mammal[") else row for row in csv.reader(fin)) 

注意：似乎在第二列中有一個前導空格。在這種情況下，在搜索/替換字符串前添加一個空格。