1. 程式人生 > >python列出指定資料夾下所有給定字尾名的檔案

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