Python列出文件夹下某类文件名的方法

Python 列出文件夹下某类文件后缀

读某个文件夹下所有图片

利用os.listdir()和string.endswith()函数实现。

在进行视觉相关任务编程时， 常常需要读出文件夹下的图片。但有的时候会包含其他后缀的文件，这时候需要将特定后缀的文件名依次读出。

python的字符串提供了一个匹配结尾的函数string.endswith(),其用法如下：
string.endswith(value, start, end)

参数	含义
value	字符串
start	检测的起始位置
end	检测的终止位置
返回值	True/False

如果需要读入的图片为xxx.jpg那么就可以只用上述函数来判断，下面是一个具体的使用例子：

#从文件夹中读取所有的图片
import os
import cv2

path = 'where/your/photo/stored'
names = os.listdir(path)    #这将返回一个所有文件名的列表
for name in names:
    if name.endswith('.jpg'):    #如果以图片jpg结尾则为真，同样对于png,bmp,txt也适用
        img = cv2.imread(path+name)
        # Your process code

然后就可以愉快地读入了~~~

icon from easyicon.net