使用Python遍歷文件夾實(shí)現(xiàn)查找指定文件夾

更新時(shí)間：2022年07月28日 16:28:06 作者：將沖破艾迪i

這篇文章主要為大家介紹了如何使用Python遍歷文件夾從而實(shí)現(xiàn)查找指定文件夾下所有相同名稱的文件、所有相同后綴名的文件，感興趣的可以了解一下

1. 文件夾結(jié)構(gòu)

指定文件夾：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)

執(zhí)行結(jié)果：

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): # 如果是目錄則遞歸，繼續(xù)查找該目錄下的文件
            search_file(absPath, fileName)
        elif currentFile == fileName:
            print(absPath) # 文件存在，則打印該文件的絕對路徑

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

執(zhí)行結(jié)果：

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)

執(zhí)行結(jié)果：

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

以下是需要注意的避坑點(diǎn)：

1.注意避坑 - 遞歸函數(shù)的位置：查找相同后綴名的文件時(shí)，遞歸調(diào)用必須放在if里，而不能放在elif里，否則當(dāng)文件夾名稱與文件后綴名相同時(shí)，會(huì)出錯(cuò)。

如：查找txt后綴的文件，剛好又有一個(gè)txt的文件夾，因?yàn)檫f歸放在了elif里，那么遍歷會(huì)停在txt文件夾，而不去遍歷txt文件夾里的內(nèi)容。

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)

執(zhí)行結(jié)果：

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打開文件夾時(shí)，會(huì)自動(dòng)生成.vscode文件夾，如果查找文件的時(shí)候不想查找這些配置文件夾里的文件，那么需要在判斷文件夾后，再加一層判斷，遇到.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)

執(zhí)行結(jié)果：