Python寫程序的時(shí)候,有時(shí)對(duì)程序進(jìn)行測試,不管你是讀取文本還是圖像,小至5、6個(gè)你可以手動(dòng)輸入,可當(dāng)有成千上萬個(gè)測試樣本的時(shí)候…….
直接進(jìn)入主題,下面簡單介紹一下 os.listdir() 這個(gè)函數(shù):
文件內(nèi)容如下:
大家可以動(dòng)手敲一下,指定一個(gè)文件夾,它會(huì)讀取所有文件的名字(但是對(duì)于這個(gè)文件夾里面的文件夾,只會(huì)列出名字,不會(huì)對(duì)讀取包含文件的里的內(nèi)容),這里大家可以自己測試下,我就不做贅述了。
讀取文件名之后,接下來要做的就是篩選出你需要你文件,下面是個(gè)簡單選出’.jpg’格式圖像的文件名:
#coding=utf-8'''使用函數(shù)os.dirlist()讀取文件夾內(nèi)的文件名字2015.08.06'''from os import listdir#將文件夾內(nèi)的文件名讀進(jìn)列表mfilepath='C:\Users\Administrator\Desktop\mycode\cv-python-day13'filename_list=listdir(filepath)h=[]#可以同過簡單后綴名判斷,篩選出你所需要的文件(這里以.jpg為例)for filename in filename_list:#依次讀入列表中的內(nèi)容 if filename[-3:]=='jpg':#將讀取文件名字的后三個(gè)字符與'jpg'匹對(duì) h.append(x)#如果是'jpg'文件就添加進(jìn)列表hprint h
這樣就能提取你需要的文件了,然后就可以逐個(gè)讀取,輸入到你的程序了。
聯(lián)系客服