Python文件重命名

时间:2013-10-27 22:47:19

标签: python

我在目录中有一些文件,

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的新手,我不确定如何实现它。

1 个答案:

答案 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