1. 程式人生 > >Python—提取資料夾下所有檔案

Python—提取資料夾下所有檔案

提取資料夾下所有檔案,儲存在L列表中,不要字尾名

import os

def file_name(file_dir):
    '''
    儲存目錄下的所有yml檔案到L列表
    '''
    L=[]   
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if os.path.splitext(file)[1] == '.yml':  
                L.append(file[:-4])   #刪除字尾名,如果需要,則 L.append(file) 
    return L  

L = file_name('E://')