前阵子,拍了200张照片,然后挑了20张找人精修下,今天修完发来给我,然后我先对比一下和原图的效果差异,以便做验收。
由于前期挑出来的照片没有留档,也就是说,还是在200张的源文件夹里,怎么从这200张图片中找出这20张呢?
幸运的是,精修完的图没有修改文件名,能够通过文件名去源文件夹找出来。那么,如何才能快速地找出这20张照片呢?
一张张搜呗!不行,太费事了!
那能不能让Python 来搭把手呢?当然可以!
环境说明:Python 3.9.12、windows11 64位,该代码应该通用,是一些内置库。
其实很简单,前面也介绍了:就是拿精修图的文件名去源文件夹中找,然后把文件移动到指定的文件夹。
拆分步骤来看就是:获取精修图的名称->在源文件夹查找->将源文件找到的图片放到指定的文件夹中。
获取精修的20张图的名称,可以使用os.listdir(path)
直接全部取出来,只需要传入对应的path
即可,这个比较简单。
假设精修图的路径是D:\精修图
,则代码可以这么写:
import os
# 精修图路径
trim_catalog = r'D:\精修图'
# 读取所有文件
trim_files = os.listdir(trim_catalog)
print(trim_files)
怎么在源文件查找图片呢?
有一种方案就是先把源文件夹的 200 张图片的名称也读取出来,然后逐一和精修图的名称进行匹配,匹配上了就返回。
这是比较常规的思路,有点像模拟在文件夹中搜索的过程。(示例图片是 AI 画的,非实际操作图)
这种方案固然可以,但是感觉不够优雅,有没有更好的方案呢?
下面是我的处理方案:
假设源文件夹的路径是D:\源文件夹
。
其实没有必要去搜索,直接将路径拼接起来,然后传递到后面的环节即可。
import os
# 精修图的名称
trim_file = '1.png'
# 源文件夹路径
src_catalog = r'D:\源文件夹'
# 拼接路径
src_file = os.path.join(src_catalog, trim_file)
print(src_file)
这样可能会出现一个问题就是图片不存在导致抛出异常,不过这没关系,既然都能预料到,那就加一层报错处理即可(后面说明)。
将图片移动到目标文件夹,这个可以通过shutil.move(old_file, new_file)
实现。
了解了这个库和函数之后,就很简单了。
假设目标存放精修图的原图的路径是D:\源文件夹\目标图
,则代码如下:
import shutil, os
# 目标目录
dst_catalog = r'D:\源文件夹\目标图'
# 精修图的名称
trim_file = '1.png'
# 拼接路径,精修图原图新的存放路径
dst_file = os.path.join(dst_catalog, trim_file)
# 移动文件,src_file 是精修图原图在源文件夹中的存放路径,在上一小节(源文件夹查找图片)中的拼接的路径
shutil.move(src_file, dst_file)
至此,三段小代码都完成了,合并起来就是:
注:将trim_file = '1.png'
改为trim_file = trim_files[0]
# 获取精修图名称
import os
# 精修图路径
trim_catalog = r'D:\精修图'
# 读取所有文件
trim_files = os.listdir(trim_catalog)
print(trim_files)
# 源文件夹查找图片
import os
# 精修图的名称
trim_file = trim_files[0]
# 源文件夹路径
src_catalog = r'D:\源文件夹'
# 拼接路径
src_file = os.path.join(src_catalog, trim_file)
print(src_file)
# 将图片放到指定文件夹
import shutil, os
# 目标目录
dst_catalog = r'D:\源文件夹\目标图'
# 精修图的名称
trim_file = trim_files[0]
# 拼接路径,精修图原图新的存放路径
dst_file = os.path.join(dst_catalog, trim_file)
# 移动文件,src_file 是精修图原图在源文件夹中的存放路径,在上一小节(源文件夹查找图片)中的拼接的路径
shutil.move(src_file, dst_file)
先测试一下上面的代码,顺利的情况下,以上代码就可以完成第一张图片的移动任务了。
那要完成 20 张,就需要对精修图进行一次遍历。
加个 for 循环,并将需要循环执行的代码都整合到循环中,放到 for 循环的代码比较简单,将拼接和移动的代码放进去即可,如下:
import shutil, os
# 精修图路径
trim_catalog = r'D:\精修图'
# 读取所有精修图文件
trim_files = os.listdir(trim_catalog)
# 源文件夹路径
src_catalog = r'D:\源文件夹'
# 目标目录
dst_catalog = r'D:\源文件夹\目标图'
for trim_file in trim_files:
# 拼接路径,精修图原图 旧 的存放路径
src_file = os.path.join(src_catalog, trim_file)
# 拼接路径,精修图原图 新 的存放路径
dst_file = os.path.join(dst_catalog, trim_file)
# 移动文件
shutil.move(src_file, dst_file)
理想情况下,以上代码便可完成任务了,但是不够强壮!前面也讲到了,可能会出现图片不存在等可能报错的情况,为了避免这种情况,需要加个try…except…
进行错误判断。
import shutil, os
# 精修图路径
trim_catalog = r'D:\精修图'
# 读取所有精修图文件
trim_files = os.listdir(trim_catalog)
# 源文件夹路径
src_catalog = r'D:\源文件夹'
# 目标目录
dst_catalog = r'D:\源文件夹\目标图'
for trim_file in trim_files:
# 拼接路径,精修图原图 旧 的存放路径
src_file = os.path.join(src_catalog, trim_file)
# 拼接路径,精修图原图 新 的存放路径
dst_file = os.path.join(dst_catalog, trim_file)
try:
# 移动文件
shutil.move(src_file, dst_file)
print('完成 %s 文件的移动。' % trim_file)
except:
print('文件 %s 移动失败!!!' % trim_file)
这样子可看到哪一些图片执行成功,哪一些执行失败,然后对执行失败的再进行查看是不是文件不存在等情况进行处理。但是这样要一个个找,能不能再给一个报告之类的呢,再整一下!
import shutil, os
# 精修图路径
trim_catalog = r'D:\精修图'
# 读取所有精修图文件
trim_files = os.listdir(trim_catalog)
# 源文件夹路径
src_catalog = r'D:\源文件夹'
# 目标目录
dst_catalog = r'D:\源文件夹\目标图'
suc_no = 0
err_no = 0
err_file_name = []
for trim_file in trim_files:
# 拼接路径,精修图原图 旧 的存放路径
src_file = os.path.join(src_catalog, trim_file)
# 拼接路径,精修图原图 新 的存放路径
dst_file = os.path.join(dst_catalog, trim_file)
try:
# 移动文件
shutil.move(src_file, dst_file)
print('完成 %s 文件的移动。' % trim_file)
suc_no += 1
except:
print('文件 %s 移动失败!!!' % trim_file)
err_no += 1
err_file_name.append(trim_file)
if err_no>0:
print('最终结果:成功移动 %s 个文件,有 %s 个文件移动失败。失败文件为 %s。' % (suc_no, err_no, '、'.join(err_file_name)))
else:
print('最终结果:成功移动 %s 个文件,无失败任务。' % suc_no)
然后复制三张放到【精修图】(假设已精修)
下面我把【精修图】中的 7.png 修改为 7-1.png ,并将移动到【目标图】的三张图重新放回【源文件夹】,再试试效果
执行结果如下:
完美!!!
本文通过 Python 的 os 和 shutil 库,使用os.listdir()
、os.path.join()
、shutil.move()
等方法,实现了将指定的文件从大文件夹中移动到目标文件夹中。
os.listdir()
用于提取指定路径的文件的名称;os.path.join()
用于拼接路径,以便通过绝对路径对文件进行移动;shutil.move()
用于移动文件。- End -