python遍历到文件夹的根目录查找图像
时间: 2023-08-08 13:01:54 浏览: 113
在Python中,我们可以使用模块来遍历文件夹并查找图像。要查找文件夹中的图像,需要使用`os`和`glob`这两个模块。
首先,我们需要导入这两个模块:
```python
import os
import glob
```
然后,我们定义一个函数来遍历文件夹并查找图像:
```python
def find_images(root_dir):
for root, dirs, files in os.walk(root_dir):
for file in files:
if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"):
image_path = os.path.join(root, file)
# 这里可以对找到的图像路径进行处理,比如打印、保存等
print(image_path)
```
在这个函数中,我们使用`os.walk`函数来遍历文件夹。`os.walk`返回的是一个生成器(generator),它会递归地遍历指定文件夹下的所有子文件夹。
在`os.walk`的循环中,我们使用`glob`模块来筛选文件后缀为`.jpg`、`.png`或`.jpeg`的文件。如果找到符合条件的文件,我们就可以对其路径进行处理,比如打印、保存等。
最后,我们可以调用这个函数来查找文件夹中的图像:
```python
find_images("/path/to/root_dir")
```
其中,`/path/to/root_dir`是你要遍历的文件夹的根目录路径。
通过这个函数,我们可以遍历文件夹的根目录并查找图像文件,并对找到的图像路径进行相应的处理。
阅读全文