使用 Python 在目录中列出特定类型的全部文件

---

Python 的灵活特性和强大的库使得操作文件和目录变得轻而易举。Python 可以帮助您生成、更新和从文件和文件夹中检索数据。其中一个典型的需求是列出特定类型的所有文件的目录列表。本教程将引导您完成该过程，并使用实际示例演示 Python 在文件系统操作方面的熟练程度。

Python 的 Os 和 Glob 库简介

标准 Python 库包含许多可以管理文件系统操作的模块。os 和 glob 模块是两个众所周知的模块示例。

• os 模块 - Python 的 os 模块提供了与操作系统通信的工具。它提供了添加、删除和浏览目录的操作，这对于文件和目录操作非常有用。

• glob 模块− 另一个有效的工具是 glob 模块，它使用 Unix shell 的规则来查找与特定模式匹配的所有路径名。鉴于它支持 * 和 ? 等通配符，因此非常适合我们的目标。

让我们深入研究示例以更好地了解它们的用法。

示例 1：使用 Os 和 Fnmatch 模块

在这里，我们将使用 Python 中的 os 和 fnmatch 模块列出目录中存在的每个 .txt 文件。

import os
import fnmatch

def list_files(directory, filetype):
   for file in os.listdir(directory):
      if fnmatch.fnmatch(file, filetype):
         print(file)

list_files('/home/user/documents', '*.txt')

此代码段中的 os.listdir() 函数显示指定目录中每个文件和目录的列表。然后，fnmatch 模块的 fnmatch() 函数确定每个文件是否与指定的文件类型模式匹配。如果文件类型匹配，则打印文件名。

示例 2：使用 Glob 模块

现在让我们使用 glob 模块列出目录中存在的每个 .jpg 图像文件。

import glob

def list_files(directory, filetype):
   for file in glob.glob(f"{directory}/{filetype}"):
      print(file)

list_files('/home/user/images', '*.jpg')

此示例中使用 glob.glob() 函数返回与路径名模式匹配的路径列表。目录路径和文件类型在 f 字符串中组合。然后，该函数打印出目录中的 .jpg 文件。

示例 3：使用 Glob 模块进行递归搜索

使用 glob 模块，我们还可以对目录及其子目录中特定类型的文件进行递归搜索。例如，让我们查找所有 .png 图片文件。

import glob

def list_files(directory, filetype):
   for file in glob.glob(f"{directory}/**/{filetype}", recursive=True):
      print(file)

list_files('/home/user/images', '*.png')

此代码段的路径字符串中的 ** 指示 glob 在所有文件夹和子目录中执行递归搜索。递归功能通过 recursive=True 选项启用。

示例 4：列出所有 Python 文件

可以按以下方式使用 os 和 fnmatch 模块来查找并列出特定目录中的每个 Python .py 文件 -

import os
import fnmatch

def list_files(directory, filetype):
   for file in os.listdir(directory):
      if fnmatch.fnmatch(file, filetype):
         print(file)

list_files('/home/user/my_python_projects', '*.py')

这将列出目录中所有 .py 文件的名称。

示例 5：使用 Os.walk 进行递归搜索

此外，我们可以使用 os.walk() 执行递归搜索。此函数自上而下或自下而上遍历目录树以创建文件名。

import os
import fnmatch

def list_files(directory, filetype):
   for dirpath, dirnames, files in os.walk(directory):
      for file in files:
         if fnmatch.fnmatch(file, filetype):
            print(os.path.join(dirpath, file))

list_files('/home/user/projects', '*.txt')

此脚本使用 os.walk() 函数遍历目录树。使用 fnmatch.fnmatch() 函数将每个文件与所需的文件类型进行比较，并使用 os.path.join() 获取匹配文件的完整路径。

结论

许多编码场景都需要在目录中列出特定类型的全部文件，而 Python 的广泛模块集（例如 os 和 glob）提供了一种简单的方法来执行此操作。本文提供的众多实际示例使如何使用这些模块在目录中列出文件以及对目录和任何子目录执行递归搜索变得一目了然。

在您的编程工具箱中具备操作文件和目录的能力至关重要。这些能力有助于组织项目文件、浏览日志文件，甚至自动化系统功能。继续研究这些例程，以发现 Python 强大的文件系统管理功能的更多内容。