本文采用os.walk()和os.listdir()兩種方法，獲取指定資料夾下的檔名。

os.walk()

模組os中的walk()函式可以遍歷資料夾下所有的檔案。

os.walk(top, topdown=Ture, onerror=None, followlinks=False)  

該函式可以得到一個三元tupple(dirpath, dirnames, filenames).

引數含義：
- dirpath：string，代表目錄的路徑；
- dirnames：list，包含了當前dirpath路徑下所有的子目錄名字（不包含目錄路徑）；
- filenames：list，包含了當前dirpath路徑下所有的非目錄子檔案的名字（不包含目錄路徑）。

注意，dirnames和filenames均不包含路徑資訊，如需完整路徑，可使用os.path.join(dirpath, dirnames)

下面給出程式碼：

# -*- coding: utf-8 -*-   

import os  

def file_name(file_dir):   
    for root, dirs, files in os.walk(file_dir):  
        print(root) #當前目錄路徑  
        print(dirs) #當前路徑下所有子目錄  
        print(files) #當前路徑下所有非目錄子檔案  
 

當需要特定型別的檔案時，程式碼如下：

# -*- coding: utf-8 -*-   

import os  

def file_name(file_dir):   
    L=[]   
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if os.path.splitext(file)[1] == '.jpeg':  
                L.append(os.path.join(root, file))  
    return
 
 L  

其中os.path.splitext()函式將路徑拆分為檔名+副檔名，例如os.path.splitext(“E:/lena.jpg”)將得到”E:/lena“+”.jpg”。

os.listdir()

os.listdir()函式得到的是僅當前路徑下的檔案名，不包括子目錄中的檔案，所有需要使用遞迴的方法得到全部檔名。

直接給出程式碼，函式將返回型別為‘.jpeg’個檔名：

# -*- coding: utf-8 -*-  
import os  

def listdir(path, list_name):  
    for file in os.listdir(path):  
        file_path = os.path.join(path, file)  
        if os.path.isdir(file_path):  
            listdir(file_path, list_name)  
        elif os.path.splitext(file_path)[1]=='.jpeg':  
            list_name.append(file_path)  

參考這裡