Python寫程序的時(shí)候，有時(shí)對(duì)程序進(jìn)行測(cè)試，不管你是讀取文本還是圖像，小至5、6個(gè)你可以手動(dòng)輸入，可當(dāng)有成千上萬個(gè)測(cè)試樣本的時(shí)候…….

直接進(jìn)入主題，下面簡(jiǎn)單介紹一下 os.listdir() 這個(gè)函數(shù)：

文件內(nèi)容如下：

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

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

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

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

h=[]

#可以同過簡(jiǎn)單后綴名判斷，篩選出你所需要的文件(這里以.jpg為例)
for filename in filename_list:#依次讀入列表中的內(nèi)容
    if filename[-3:]=='jpg':#將讀取文件名字的后三個(gè)字符與'jpg'匹對(duì)
        h.append(x)#如果是'jpg'文件就添加進(jìn)列表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

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