python如何读取文件夹图片
时间: 2024-06-12 22:11:34 浏览: 203
Python可以使用os库中的listdir()函数或者os.walk()函数来读取文件夹中的图片。其中listdir()函数可以直接获取指定文件夹下的所有文件名称,而os.walk()函数可以递归获取指定文件夹及其子文件夹下的所有文件名称。以下是两种方法的示例代码:
方法一:
```python
import os
def getfiles(input_path):
path_list = []
filenames = os.listdir(input_path)
for filename in filenames:
if filename.endswith('.jpg') or filename.endswith('.png'): # 只获取图片文件
a = os.path.join(input_path, filename)
path_list.append(a)
return path_list
```
方法二:
```python
import os
def get_files(input_path):
path_list = []
for filepath, dirnames, filenames in os.walk(input_path):
for filename in filenames:
if filename.endswith('.jpg') or filename.endswith('.png'): # 只获取图片文件
a = os.path.join(filepath, filename)
path_list.append(a)
return path_list
```
以上两种方法都可以获取指定文件夹下的所有图片文件路径,并将其存储在一个列表中返回。需要注意的是,这里只获取了以.jpg或.png结尾的文件,如果需要获取其他格式的文件,可以根据需要修改代码。
阅读全文