使用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 模块，我们还可以在目录及其子目录中递归搜索特定类型的文件。例如，我们来查找 all.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 强大的文件系统管理功能。