Python获取指定文件夹下文件名的两种方法：os.walk与os.listdir

本文介绍如何利用Python的os模块，通过os.walk()和os.listdir()两种函数来获取指定文件夹下的文件名。

一、使用os.walk()方法

os模块中的walk()函数支持对文件夹及其子目录进行递归遍历，从而获取所有文件。

os.walk(top, topdown=Ture, onerror=None, followlinks=False) 

调用该函数会返回一个包含三个元素的元组：(dirpath, dirnames, filenames)。

各参数的具体含义如下：

1. dirpath：字符串类型，表示当前正在遍历的目录路径；
2. dirnames：列表类型，包含当前dirpath路径下所有子目录的名称（不含完整路径）；
3. filenames：列表类型，包含当前dirpath路径下所有非目录文件（即子文件）的名称（不含完整路径）。

需要注意的是，dirnames和filenames中存储的均不包含路径信息。若需获取完整文件路径，可结合使用os.path.join(dirpath, filename)。

以下是相关代码示例：

# -*- 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) #当前路径下所有非目录子文件 

若需筛选特定类型的文件，可参考以下代码逻辑：

# -*- coding: utf-8 -*-   
import os 
 
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 

代码中使用了os.path.splitext()函数，该函数用于将路径拆分为文件名和扩展名两部分。例如，对“E:/lena.jpg”执行该函数，将返回“E:/lena”和“.jpg”。

二、使用os.listdir()方法

os.listdir()函数仅返回当前指定路径下的文件名，不包含子目录中的文件。因此，若要获取所有层级的文件名，必须采用递归调用的方式。

直接提供实现代码，该函数用于返回类型为‘.jpeg’的文件名列表：

# -*- coding: utf-8 -*- 
import os 
 
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(file_path, list_name) 
    elif os.path.splitext(file_path)[1]=='.jpeg': 
      list_name.append(file_path) 

作为Python初学者，在编写代码时容易遇到一些基础问题，导致耗时较多。首先，务必确保代码遵循严格的四空格缩进规范，直接按回车换行并不会自动产生四空格缩进；其次，在Windows命令提示符中使用Python解释器时，若修改源代码后再次运行，需先退出当前解释器并重新进入，否则系统仍会执行第一次运行过的代码缓存。

以上就是本文的全部内容，希望对各位的学习有所帮助，也欢迎大家继续支持本站。