【Python】使用python遍历文件夹查找指定文件夹下所有同名文件和所有同后缀文件

文章目录

  • 1. 文件夹结构
  • 2. 查找指定文件夹下指定文件
  • 3. 查找指定文件夹下所有相同名称的文件
  • 4. 查找指定文件夹下所有相同后缀名的文件
  • 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
    
    物联沃分享整理
    物联沃-IOTWORD物联网 » 【Python】使用python遍历文件夹查找指定文件夹下所有同名文件和所有同后缀文件

    发表评论