python列出指定文件夹下所有给定后缀名的文件

写了一个小程序来实现这个功能。

程序的输入是指定的文件夹路径和指定的后缀名（默认为列出所有文件）

代码如下：

#coding:utf8
import os

def listfile(dirname,postfix = ''):
    filelist = []
    files = os.listdir(dirname)
    for item in files:
        #filelist.append([dirname,item])
        if os.path.isfile(dirname+item):
            if item.endswith(postfix):
                filelist.append([dirname,item])
        else:
            if os.path.isdir(dirname+item):
                filelist.extend(listfile(dirname+item+'/',postfix))
    return filelist