python自动化，读取文件夹内所有文件名字，并做简单的后缀名匹配

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

这样就能提取你需要的文件了，然后就可以逐个读取，输入到你的程序了。