python之获取目录下的文件名

背景：为了统计一个目录下所有文件名，并且写到文件中

工具：python的模块os.walk

功能：遍历目录下的所有子文件及其子文件夹，如果有子文件夹的话，会继续以子文件夹为目录递归遍历

返回：如果只有文件的话，只返回一个元组（当前目录字符串，文件夹名列表，文件名列表）；如果还有文件夹的话，返回多个元组

  

   

    

     

    
    

     

      #coding=utf-8
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def file_name(file_dir): 
     
    
   

    

     

    
    

     
    for root, dirs, files in os.walk(file_dir):
     
    
   

    

     

    
    

     
        #print(root) #当前目录路径
     
    
   

    

     

    
    

     
        #print(dirs) #当前路径下所有子目录
     
    
   

    

     

    
    

     
        print(files) #当前路径下所有非目录子文件
     
    
   

    

     

    
    

     
    return files
     
    
   

    

     

    
    

     

      '''
     
    
   

    

     

    
    

     

      def file_name(file_dir): 
     
    
   

    

     

    
    

     

       L=[] 
     
    
   

    

     

    
    

     

       for root, dirs, files in os.walk(file_dir):
     
    
   

    

     

    
    

     

       for file in files:
     
    
   

    

     

    
    

     

       if os.path.splitext(file)[1] == '.jpeg':
     
    
   

    

     

    
    

     

       L.append(os.path.join(root, file))
     
    
   

    

     

    
    

     

       return L
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

      def listdir(path, list_name):
     
    
   

    

     

    
    

     

       for file in os.listdir(path):
     
    
   

    

     

    
    

     

       file_path = os.path.join(path, file)
     
    
   

    

     

    
    

     

       if os.path.isdir(file_path):
     
    
   

    

     

    
    

     

       listdir(f
     
    
  
 

文章来源: wenyusuran.blog.csdn.net，作者：文宇肃然，版权归原作者所有，如需转载，请联系作者。

原文链接：wenyusuran.blog.csdn.net/article/details/82348286