1. 程式人生 > >python 返回指定目錄下的所有檔案與特定檔案

python 返回指定目錄下的所有檔案與特定檔案

有時要讀取一個資料夾下的多個檔案,下面這段程式碼即是返回指定檔案目錄下的所有檔案的檔名,存於列表中,然後從列表中進行匹配檔名中含有'.csv'的檔名,最後返回所有'.csv'檔案

讀取'.csv'檔案可以參考:

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import sys
import os
import re
def eachFile(filepath):
    pathDir =  os.listdir(filepath)  #返回指定路徑下所有檔案和資料夾的名字,並存放於一個列表中
    dir_list = []
    for allDir in pathDir:
        if re.search('.csv', allDir) != None:  #search()會掃描整個string查詢匹配,會掃描整個字串並返回第一個成功的匹配
            dir_list.append(allDir)
    return dir_list

if __name__ == '__main__':
    filepath = u"G:\\研究\\AWS\\AWS\\mod\\"  #指定檔案目錄
    dir_list = eachFile(filepath)  #獲取檔名列表
    print(dir_list)