在处理机器学习的图片时,需要将图片根据类别放入不同文件夹中,在文件夹下也想将图片用数字递增重命名一下,方便后续处理。
首先,我们看一下python的重命名函数:
os.rename(src,dst)
src:要修改的目录名
dst:修改后的目录名
如果dst是一个存在的目录, 将抛出OSError。
注意:src和dst都是完整路径+文件名
首先,我们先对文件夹重命名
先使用最简单的方法尝试一下:
import os
def myrename(path):
file_list=os.listdir(path)
i=0
for fi in file_list:
old_name=os.path.join(path,fi)
new_name=os.path.join(path,str(i))
os.rename(old_name,new_name)
i+=1
if __name__=="__main__":
path="D:/test/121"
myrename(path)
我们就可以看到文件夹已经按照顺序重命名了,稍微改动一下,显得没有那么low:
import os
def myrename(path):
file_list=os.listdir(path)
for i,fi in enumerate(file_list):
old_name=os.path.join(path,fi)
new_name=os.path.join(path,"N0."+str(i))
os.rename(old_name,new_name)
if __name__=="__main__":
path="D:/test/121"
myrename(path)
效果是一样的,enumerate函数的作用就不再赘述了,我们还可以使用python zip来进行修改:
import os
def myrename(path):
file_list=os.listdir(path)
for i,fi in zip(range(len(file_list)),file_list):
old_name=os.path.join(path,fi)
new_name=os.path.join(path,"The."+str(i))
os.rename(old_name,new_name)
if __name__=="__main__":
path="D:/test/121"
myrename(path)
效果一样:
接下来我们对文件进行重命名,上面的三种方法都是可以使用的,只需要稍微改动一下,我以函数enumerate为例演示一下:
先建一个文本文档,复制粘贴一堆,作为测试用,在函数中加入一个try-except,防止文件重命名出错,这里重命名需要是相同格式的文件,否则就会报错:
import os
def myrename(path):
file_list=os.listdir(path)
for i,fi in enumerate(file_list):
old_dir=os.path.join(path,fi)
filename="my"+str(i+1)+"."+str(fi.split(".")[-1])
new_dir=os.path.join(path,filename)
try:
os.rename(old_dir,new_dir)
except Exception as e:
print(e)
print("Failed!")
else:
print("SUcess!")
if __name__=="__main__":
path="D:/test/121"
myrename(path)