- 需求
首先说一下我的需求,我有一个文件夹里面有多个子文件夹,每个子文件夹里面有不同数量的图片,我需要把这些图片全部放在同一个文件夹里面,并统计全部图片数量。 - 环境说明
Win10 系统下 Python3,编译器是 Pycharm。
需要shutil第三方库,直接pip install shutil会报错
C:\Users\Administrator>pip install shutil
ERROR: Could not find a version that satisfies the requirement shutil (from versions: none)
ERROR: No matching distribution found for shutil
正确的安装方法如下:
pip install pytest-shutil
- 代码
先导入需要用到的库,os 是用来切换路径和创建文件夹的。
shutil 是用来复制黏贴文件的:
import os
from shutil import copy
i 用来统计图片的数量
save_dir 是把复制出来的文件存放的路径
# i 用来统计图片数量
i = 0
# 想保存到的根路径
save_dir = r'G:\camelyon\labels'
# 如果目录不存在,则创建
if not os.path.isdir(save_dir):
os.makedirs(save_dir)
file_path 是原文件夹所在路径,也就是一大堆图片所在的路径。
os.listdir(file_path) 是获取指定路径下包含的文件或文件夹列表,在 Unix, Windows 下使用。
file_path = r'G:\病理图数据集\labels'
# 获取 file_path 下的文件夹列表
pathDir = os.listdir(file_path)
获取到了列表,那就一个一个进行遍历。
allDir 是文件的名字+后缀名。from_path 就是当前这个被遍历出来的文件的完整路径,to_path是这个文件要复制到的路径。
接着 copy(from_path, to_path) 完成复制黏贴,最后 i 自增1。
for allDir in pathDir:
print(allDir)
from_path = os.path.join(file_path, allDir)
to_path = save_dir
# 如果 to_path 目录不存在,则创建
if not os.path.isdir(to_path):
os.makedirs(to_path)
# 子文件夹列表
filelist = os.listdir('G:\病理图数据集\labels\\' + allDir)
print(filelist)
for index, files in enumerate(filelist):
copy(from_path + '\\' + files, to_path)
# 需要将files改为绝对路径
i += 1
print(i)
- 运行结果截图
原来文件夹中每个子文件夹如下
子文件夹下的图片列表
移动后的图片列表
参考链接:https://www.jianshu.com/p/acf27824e901