python 递归搜索指定文件夹下所有图片并返回图片路径
时间: 2023-12-27 10:05:13 浏览: 161
可以使用os模块和递归函数来实现。首先,需要在Python脚本中导入os模块:
```
import os
```
然后,编写一个递归函数来搜索指定文件夹下所有图片并返回图片路径:
```
def find_images(folder_path):
image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp') # 定义图片扩展名
image_paths = [] # 存储所有图片的路径
for filename in os.listdir(folder_path):
file_path = os.path.join(folder_path, filename)
if os.path.isdir(file_path):
# 如果是文件夹,则递归搜索子文件夹
image_paths += find_images(file_path)
elif os.path.isfile(file_path) and filename.lower().endswith(image_extensions):
# 如果是图片文件,则添加到路径列表中
image_paths.append(file_path)
return image_paths
```
该函数接受一个文件夹路径作为参数,并返回该文件夹下所有图片的路径。函数遍历该文件夹下的所有文件和子文件夹,并检查每个文件是否是一个图片文件。如果是,就将其路径添加到列表中。如果是文件夹,则递归搜索该文件夹下的子文件夹。
使用该函数的示例代码如下:
```
folder_path = 'path/to/folder' # 指定文件夹路径
image_paths = find_images(folder_path)
for image_path in image_paths:
print(image_path)
```
运行该代码会输出所有图片的路径。
阅读全文