我正在尝试根据csv文件中的字符串值添加计数。
我有4种不同类型的字符串值:
cat
dog
mouse
hamster
下面的代码添加count,而不考虑字符串值。
我如何让我的代码按顺序添加计数,例如cat1、cat2、dog1、dog2等。
我使用以下代码:
import os
import csv
count = 0
with open('rename.csv') as f:
lines = csv.reader(f)
for line in lines:
os.rename(line[0], line[1] + str(count) + '.jpg')
count += 1
这里有一个可能的解决方案:
from collections import Counter
import os
import csv
c = Counter()
with open('rename.csv') as f:
lines = csv.reader(f)
for line in lines:
c.update([line[1]])
os.rename(line[0], line[1] + str(c[line[1]]) + '.jpg')