首頁 > 軟體

使用Python遍歷資料夾實現查詢指定資料夾

2022-07-28 22:02:56

1. 資料夾結構

指定資料夾:E:/Code/Python/test

指定檔案:test.txt

指定資料夾下的目錄及檔案:

資料夾a:

a.txt

test.txt

資料夾txt:

test.txt

檔案b.txt

test.py

test.txt

2. 查詢指定資料夾下指定檔案

import os

# 查詢該層資料夾下指定檔案
def search_file(dirPath, fileName):
    dirs = os.listdir(dirPath) # 查詢該層資料夾下所有的檔案及資料夾,返回列表
    for currentFile in dirs: # 遍歷列表
        absPath = dirPath + '/' + currentFile # 檔案的絕對路徑
        if currentFile == fileName:
             print(absPath) 
           
if __name__ == "__main__":
    dirPath = 'E:/Code/Python/test'
    fileName = 'test.txt'
    search_file(dirPath, fileName)
              

執行結果:

E:/Code/Python/test/test.txt

3. 查詢指定資料夾下所有相同名稱的檔案

import os

# 查詢指定資料夾下所有相同名稱的檔案
def search_file(dirPath, fileName):
    dirs = os.listdir(dirPath) # 查詢該層資料夾下所有的檔案及資料夾,返回列表
    for currentFile in dirs: # 遍歷列表
        absPath = dirPath + '/' + currentFile 
        if os.path.isdir(absPath): # 如果是目錄則遞迴,繼續查詢該目錄下的檔案
            search_file(absPath, fileName)
        elif currentFile == fileName:
            print(absPath) # 檔案存在,則列印該檔案的絕對路徑

if __name__ == "__main__":
    dirPath = 'E:/Code/Python/test'
    fileName = 'test.txt'
    search_file(dirPath, fileName)

執行結果:

E:/Code/Python/test/a/test.txt
E:/Code/Python/test/test.txt    
E:/Code/Python/test/txt/test.txt

4. 查詢指定資料夾下所有相同字尾名的檔案

import os

# 查詢指定資料夾下所有相同字尾名的檔案
def search_file_suffix(dirPath, suffix):
    dirs = os.listdir(dirPath)
    for currentFile in dirs:
        absPath = dirPath + '/' + currentFile
        if os.path.isdir(absPath):
            search_file_suffix(absPath, suffix)
        elif currentFile.split('.')[-1] == suffix: # 檔案字尾名相同,則列印該檔案的絕對路徑
            print(absPath)

if __name__ == "__main__":
    dirPath = 'E:/Code/Python/test'
    suffix = 'txt'
    search_file_suffix(dirPath, suffix)
      

執行結果:

E:/Code/Python/test/a/a.txt
E:/Code/Python/test/a/test.txt  
E:/Code/Python/test/b.txt       
E:/Code/Python/test/test.txt    
E:/Code/Python/test/txt/test.txt

以下是需要注意的避坑點:

1.注意避坑 - 遞迴函數的位置:查詢相同字尾名的檔案時,遞迴呼叫必須放在if裡,而不能放在elif裡,否則當資料夾名稱與檔案字尾名相同時,會出錯。

如:查詢txt字尾的檔案,剛好又有一個txt的資料夾,因為遞迴放在了elif裡,那麼遍歷會停在txt資料夾,而不去遍歷txt資料夾裡的內容。

import os

# 查詢指定資料夾下所有相同字尾名的檔案
def search_file_suffix(dirPath, suffix):
    dirs = os.listdir(dirPath)
    for currentFile in dirs:
        absPath = dirPath + '/' + currentFile
        if os.path.isdir(absPath):
            search_file_suffix(absPath, suffix)
        elif currentFile.split('.')[-1] == suffix: # 檔案字尾名相同,則列印該檔案的絕對路徑
            print(absPath)

def search_file_suffix_temp(dirPath, suffix):
    dirs = os.listdir(dirPath)
    for currentFile in dirs:
        absPath = dirPath + '/' + currentFile
        if currentFile.split('.')[-1] == suffix: 
            print(absPath)
        elif os.path.isdir(absPath): # 遞迴判斷放在elif裡
            search_file_suffix_temp(absPath, suffix)

if __name__ == "__main__":
    dirPath = 'E:/Code/Python/test'
    suffix = 'txt'
    search_file_suffix(dirPath, suffix)
    print("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
    search_file_suffix_temp(dirPath, suffix)

執行結果:

E:/Code/Python/test/a/a.txt
E:/Code/Python/test/a/test.txt  
E:/Code/Python/test/b.txt       
E:/Code/Python/test/test.txt    
E:/Code/Python/test/txt/test.txt
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
E:/Code/Python/test/a/a.txt     
E:/Code/Python/test/a/test.txt  
E:/Code/Python/test/b.txt       
E:/Code/Python/test/test.txt
E:/Code/Python/test/txt

2.注意避坑 - .vscode資料夾:在使用vscode開啟資料夾時,會自動生成.vscode資料夾,如果查詢檔案的時候不想查詢這些組態檔夾裡的檔案,那麼需要在判斷資料夾後,再加一層判斷,遇到.vscode或者其他組態檔夾就跳過。

import os
from fnmatch import fnmatchcase

# 查詢指定資料夾下所有相同字尾名的檔案
def search_file_suffix(dirPath, suffix):
    dirs = os.listdir(dirPath)
    for currentFile in dirs:
        absPath = dirPath + '/' + currentFile
        if os.path.isdir(absPath):
            search_file_suffix(absPath, suffix)
        elif currentFile.split('.')[-1] == suffix: # 檔案字尾名相同,則列印該檔案的絕對路徑
            print(absPath)

def search_file_suffix_temp(dirPath, suffix):
    dirs = os.listdir(dirPath)
    for currentFile in dirs:
        absPath = dirPath + '/' + currentFile
        if os.path.isdir(absPath):
            if fnmatchcase(currentFile,'.*'): # 如果資料夾是.vscode或者其他.開頭的資料夾,就跳過
                pass
            else:
                search_file_suffix(absPath, suffix)
        elif currentFile.split('.')[-1] == suffix: 
            print(absPath)

if __name__ == "__main__":
    dirPath = 'E:/Code/python/test'
    suffix = 'json'
    search_file_suffix(dirPath, suffix)
    print("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
    search_file_suffix_temp(dirPath, suffix)    

執行結果:

E:/Code/python/test/.vscode/settings.json
E:/Code/python/test/a.json
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
E:/Code/python/test/a.json

到此這篇關於使用Python遍歷資料夾實現查詢指定資料夾的文章就介紹到這了,更多相關Python查詢指定資料夾內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com