我在目录中有一些文件,
file_IL.txt
file_IL.csv
file_NY.txt
file_NY.csv
我必须重命名它们才能获得序列号。例如,
file_IL.txt_001
file_IL.csv_001
file_NY.txt_002
file_NY.csv_002
我编写了以下Python代码
def __init__(self):
self.indir = "C:\Files"
def __call__(self):
found = glob.glob(self.indir + '/file*')
length = len(glob.glob(self.indir + '/file*'))
print length
count = 000
for num in (glob.glob(self.indir + '/file*')):
count = count + 1
count = str(count)
print count
shutil.copy(num, num+'_'+count)
print num
count = int(count)
但是这给了我一个如下结果,
file_IL.txt_001
file_IL.csv_002
file_NY.txt_003
file_NY.csv_004
有人可以帮我修改上面的Python脚本以符合我的要求吗?我是Python的新手,我不确定如何实现它。
答案 0 :(得分:3)
最好的方法是在字典中存储该扩展名的扩展名和计数。
def __call__(self):
found = glob.glob(self.indir + '/file*')
length = len(found)
counts = {}
for num in found:
ext = num.rsplit(".",1)[-1] # Right split to get the extension
count = counts.get(ext,0) + 1 # get the count, or the default of 0 and add 1
shutil.copy(num, num+'_'+'%03d' % count) # Fill to 3 zeros
counts[ext] = count # Store the new count