Python教程网
--左手Python,右手AI!

我是如何用10行Python代码搬运目标图片的?

嗯呢,你没看错,就是教你把一个路径下的所有目标图片搬运到制定路径下。有读者说:小詹你忽悠人吧,要搬运目标图片复制粘贴不就好了嘛,要什么代码,搬砖脑子秀逗了?

咳咳,对于目标文件夹复制粘贴当然可以,还简单易操作。但是如果你要的目标图片在某一路径下不同深度的子文件夹中呢?如果该文件夹中各种格式的文件扎堆在一起呢?比如下面这样:

  • –dataset (文件夹)
    • –pic0.jpg
    • –file_1(子文件夹)
      • –pic1.jpg
      • –pic2.jpg
      • –file_1_1 (子文件夹的子文件夹)
        • –pic3.jpg
        • –pic4.jpg
        • –txt.txt
        • …………

如果你要搬运的图片是上边那样存放的呢?如果层数更多更复杂呢?如果图片数量庞大呢?如果其他格式的文件干扰呢?(不要觉得这现象少见,在处理数据集的时候多少都会遇到的~)复制粘贴效率就不行了吧!别忘了咱们是谁,是改变世界的程序猿啊!这不,哪怕再复杂,下边用10来行代码轻松搞定~

说的很有趣很高深似的,其实就是用了Python的两个模块os和shutil罢了,用到其中两个方法,一一道来。

1.首先是文件复制方法

shutil.copy(src,dst)

复制一个文件到另一个文件或目录,对于本例而言就是单次操作的图片。

2.还有遍历目录操作

os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]])

  • top — 是你所要遍历的目录的地址, 返回的是一个三元组(root,dirs,files)。
    • root 所指的是当前正在遍历的这个文件夹的本身的地址
    • dirs 是一个 list ,内容是该文件夹中所有的目录的名字(不包括子目录)
    • files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)
    • topdown –可选,为 True,则优先遍历 top 目录,否则优先遍历 top 的子目录(默认为开启)。如果 topdown 参数为 True,walk 会遍历top文件夹,与top 文件夹中每一个子目录。
  • onerror — 可选,需要一个 callable 对象,当 walk 需要异常时,会调用。
  • followlinks — 可选,如果为 True,则会遍历目录下的快捷方式,如果为 False,则优先遍历 top 的子目录。

弄明白这两个关键方法,要实现目标图片的搬运就只需要10行代码了~

import os,shutil 

print('输入格式:E:\myprojectnew\jupyter\整理文件夹\示例')
path = input('请键入需要整理的文件夹地址:')
new_path = input('请键入要复制到的文件夹地址:')
 
for root, dirs, files in os.walk(path):  
    for i in range(len(files)):
        if (files[i][-3:] == 'jpg') or (files[i][-3:] == 'png') or (files[i][-3:] == 'JPG'):
            file_path = root+'/'+files[i]  
            new_file_path = new_path+ '/'+ files[i]  
            shutil.copy(file_path,new_file_path)  

运行后只需要指定你想要复制的图片所在路径为path,欲复制到的路径为new_path,就可以等着自动将图片复制到指定路径下了,当然目标不限于图片噢。

3.相关补充

os模块和shutil模块常用的方法还有以下几种~

# os模块

os.sep    可以取代操作系统特定的路径分隔符。windows下为 '\\'
os.name    字符串指示你正在使用的平台。比如对于Windows,它是'nt',而对于Linux/Unix用户,它是 'posix'
os.getcwd()    函数得到当前工作目录,即当前Python脚本工作的目录路径
os.getenv()    获取一个环境变量,如果没有返回none
os.putenv(key, value)    设置一个环境变量值
os.listdir(path)    返回指定目录下的所有文件和目录名
os.remove(path)    函数用来删除一个文件
os.system(command)    函数用来运行shell命令
os.linesep    字符串给出当前平台使用的行终止符。例如,Windows使用 '\r\n',Linux使用 '\n' 而Mac使用 '\r'
os.path.split(path)        函数返回一个路径的目录名和文件名
os.path.isfile()    和os.path.isdir()函数分别检验给出的路径是一个文件还是目录
os.path.exists()    函数用来检验给出的路径是否真地存在
os.curdir        返回当前目录 ('.')
os.mkdir(path)    创建一个目录
os.makedirs(path)    递归的创建目录
os.chdir(dirname)    改变工作目录到dirname          
os.path.getsize(name)    获得文件大小,如果name是目录返回0L
os.path.abspath(name)    获得绝对路径
os.path.normpath(path)    规范path字符串形式
os.path.splitext()        分离文件名与扩展名
os.path.join(path,name)    连接目录与文件名或目录
os.path.basename(path)    返回文件名
os.path.dirname(path)    返回文件路径
os.walk(top,topdown=True,onerror=None)        遍历迭代目录
os.rename(src, dst)        重命名file或者directory src到dst 如果dst是一个存在的directory, 将抛出OSError. 在Unix, 如果dst在存且是一个file, 如果用户有权限的话,它将被安静的替换. 操作将会失败在某些Unix 中如果src和dst在不同的文件系统中. 如果成功, 这命名操作将会是一个原子操作 (这是POSIX 需要). 在 Windows上, 如果dst已经存在, 将抛出OSError,即使它是一个文件. 在unix,Windows中有效。
os.renames(old, new)    递归重命名文件夹或者文件。像rename()

# shutil 模块

shutil.copyfile( src, dst)    从源src复制到dst中去。当然前提是目标地址是具备可写权限。抛出的异常信息为IOException. 如果当前的dst已存在的话就会被覆盖掉
shutil.move( src, dst)        移动文件或重命名
shutil.copymode( src, dst)    只是会复制其权限其他的东西是不会被复制的
shutil.copystat( src, dst)    复制权限、最后访问时间、最后修改时间
shutil.copy( src, dst)        复制一个文件到一个文件或一个目录
shutil.copy2( src, dst)        在copy上的基础上再复制文件最后访问时间与修改时间也复制过来了,类似于cp –p的东西
shutil.copy2( src, dst)        如果两个位置的文件系统是一样的话相当于是rename操作,只是改名;如果是不在相同的文件系统的话就是做move操作
shutil.copytree( olddir, newdir, True/Flase)
把olddir拷贝一份newdir,如果第3个参数是True,则复制目录时将保持文件夹下的符号连接,如果第3个参数是False,则将在复制的目录下生成物理副本来替代符号连接
shutil.rmtree( src )    递归删除一个目录以及目录内的所有内容
赞(11) 打赏
未经允许不得转载:Python教程网 » 我是如何用10行Python代码搬运目标图片的?
分享到: 更多 (0)

评论 1

  1. #1

    不错,学习了

    匿名5天前回复

一块钱也是爱,支持最重要~

支付宝扫一扫打赏

微信扫一扫打赏