python获取指定目录下的所有指定后缀的文件名

代码如下

#! /usr/bin/python
# -*- coding: utf-8 -*-
import os

def getFileName(path):
    ''' 获取指定目录下的所有指定后缀的文件名 '''
    f_list = os.listdir(path)
    for i in