python 修改文件名和提取文件到txt

修改文件名：

import os
import sys
file='C:'
#i = 0
def refilename(file):
dir = os.path.abspath(file)
dirlist=os.listdir(dir)
i=0
        for dir1 in dirlist:
dir1join=os.path.join(dir,dir1)
#for i in range(len(dirlist)):
k=(i+1)*1000
if dir1join !='.' and dir1join !='..' and os.path.isdir(dir1join): 
dirlist=os.listdir(dir1join)
for filename in dirlist:
olddir=os.path.join(dir1join,filename)
oldname=os.path.splitext(filename)[0]
filetype=os.path.splitext(filename)[1]
k=k+1
s=str(k)
newdir=os.path.join(dir1join,s+filetype)
os.rename(olddir,newdir)
i=i+1


if __name__ == "__main__":
refilename('train')

refilename('test')

提取文件到txt

import os
import sys
import os.path
#dir_name = 'I:\vs2015\bofvs2010\data'
def pic_list(dir_name):
    fp = open(dir_name + '.txt', 'w')
    cdir = os.path.abspath(dir_name)
    dirs = os.listdir(cdir)
    for d in dirs:
        dir_path = os.path.join(cdir, d)
        if d != '.' and d != '..' and os.path.isdir(dir_path):
            pics = os.listdir(dir_path)
            fp.write('%s %d\n' % (str(d), len(pics)))
            for p in pics:
                fp.write('%s\n' % str(os.path.join(dir_path, p)))                
    fp.close()
                
if __name__ == '__main__':
    pic_list('train')
    pic_list('test')