在文件名后加_mask字符串
import os path = 'F:/test/msk' datanames = os.listdir(path) for i in datanames: n = i.split('.') new_i = n[0] + "_mask" # 在此处有区别:把想要增加的内容,以字符串的形式放在末尾即可 new_name = new_i + '.png' if i.endswith(".png"): # 判断是否是.jpg文件 os.rename(os.path.join(path, i), os.path.join(path, new_name))
将文件名特定字符替换,_gt替换成_mask
import os # s = 'gt' #需要替换的字段 r = 'mask' # 替换的字段 # def openNextF(paths=['F:/test/msk']): global s pathsN=[] for p in paths: # 解决当前目录,路径为空的情况 if not p: fs = os.listdir() print(fs) else: fs = os.listdir(p) # 打开文件夹名不是路径,当前相对路径 for f in fs: # 设置下一层路径 if not p: pN = f else: pN = p+'/'+f print('当前f:',pN) #判断是否是文件夹 if os.path.isfile(pN): if s in f: os.rename(pN,pN.replace(s,r)) else: pathsN.append(pN) return (pathsN) # 调用函数 paths = ['F:/test/msk'] n = 20 # 打开多少层目录 for i in range(n): paths = openNextF(paths) if len(paths)==0: break print('第%d层目录:'%(i+1)+str(paths))