Python寫程序的時候，有時對程序進行測試，不管你是讀取文本還是圖像，小至5、6個你可以手動輸入，可當有成千上萬個測試樣本的時候…….

直接進入主題，下面簡單介紹一下 os.listdir() 這個函數：

文件內容如下：

大家可以動手敲一下，指定一個文件夾，它會讀取所有文件的名字（但是對于這個文件夾里面的文件夾，只會列出名字，不會對讀取包含文件的里的內容），這里大家可以自己測試下，我就不做贅述了。

讀取文件名之后，接下來要做的就是篩選出你需要你文件，下面是個簡單選出’.jpg’格式圖像的文件名：

#coding=utf-8
'''
使用函數os.dirlist()讀取文件夾內的文件名字
2015.08.06
'''
from os import listdir

#將文件夾內的文件名讀進列表m
filepath='C:\Users\Administrator\Desktop\mycode\cv-python-day13'
filename_list=listdir(filepath)

h=[]

#可以同過簡單后綴名判斷，篩選出你所需要的文件(這里以.jpg為例)
for filename in filename_list:#依次讀入列表中的內容
    if filename[-3:]=='jpg':#將讀取文件名字的后三個字符與'jpg'匹對
        h.append(x)#如果是'jpg'文件就添加進列表h
print h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

這樣就能提取你需要的文件了，然后就可以逐個讀取，輸入到你的程序了。