Python3学习笔记09-枚举指定目录下文件名和子目录名

枚举文件名有两种方法,os.walk()和os.listdir(),如下分别介绍。

一、os.walk()

模块os中的walk()函数可以遍历文件夹下所有的文件。

[python]  view plain  copy
  1. os.walk(top, topdown=Ture, οnerrοr=None, followlinks=False)  
该函数可以得到一个三元tupple(dirpath, dirnames, filenames).

参数含义:

  • dirpath:string,代表目录的路径;
  • dirnames:list,包含了当前dirpath路径下所有的子目录名字(不包含目录路径);
  • filenames:list,包含了当前dirpath路径下所有的非目录子文件的名字(不包含目录路径)。

注意,dirnames和filenames均不包含路径信息,如需完整路径,可使用os.path.join(dirpath, dirnames)

下面给出代码;

[python]  view plain  copy
  1. # -*- coding: utf-8 -*-   
  2.   
  3. import os  
  4.   
  5. def file_name(file_dir):   
  6.     for root, dirs, files in os.walk(file_dir):  
  7.         print(root) #当前目录路径  
  8.         print(dirs) #当前路径下所有子目录  
  9.         print(files) #当前路径下所有非目录子文件  

当需要特定类型的文件时,代码如下:

[python]  view plain  copy
  1. # -*- coding: utf-8 -*-   
  2.   
  3. import os  
  4.   
  5. def file_name(file_dir):   
  6.     L=[]   
  7.     for root, dirs, files in os.walk(file_dir):  
  8.         for file in files:  
  9.             if os.path.splitext(file)[1] == '.jpeg':  
  10.                 L.append(os.path.join(root, file))  
  11.     return L  
其中os.path.splitext()函数将路径拆分为文件名+扩展名,例如os.path.splitext(“E:/lena.jpg”)将得到”E:/lena“+".jpg"。


二、os.listdir()

os.listdir()函数得到的是仅当前路径下的文件名,不包括子目录中的文件,所有需要使用递归的方法得到全部文件名。

直接给出代码,函数将返回类型为‘.jpeg’个文件名:

[python]  view plain  copy
  1. # -*- coding: utf-8 -*-  
  2. import os  
  3.   
  4. def listdir(path, list_name):  
  5.     for file in os.listdir(path):  
  6.         file_path = os.path.join(path, file)  
  7.         if os.path.isdir(file_path):  
  8.             listdir(file_path, list_name)  
  9.         elif os.path.splitext(file_path)[1]=='.jpeg':  
  10.             list_name.append(file_path)  

你可能感兴趣的:(Python编程手札,枚举文件名,dir())