python寻找指定后缀文件

原创

mob64ca12edad02 2023-10-05 16:36:32 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12edad02的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python寻找指定后缀文件

在日常的编程工作中，我们经常需要寻找某个目录下特定后缀的文件。这可能是为了查找某些特定类型的文件，或者是为了进行进一步的处理。Python提供了多种方法来实现这一目标，本文将介绍其中的几种常用方法，并提供代码示例。

方法一：使用os模块和字符串切片

首先，我们可以使用Python的os模块来遍历目录下的所有文件，并使用字符串切片来判断文件名的后缀是否符合要求。下面是示例代码：

import os

def find_files_with_extension(directory, extension):
    result = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                result.append(os.path.join(root, file))
    return result

directory = '/path/to/directory'
extension = '.txt'
files = find_files_with_extension(directory, extension)
print(files)

在这个示例中，我们定义了一个find_files_with_extension函数，它接受一个目录路径和一个后缀名作为参数，并返回符合条件的文件路径列表。我们使用os.walk函数来遍历目录下的所有文件和子目录，然后使用字符串的endswith方法来判断文件名是否以指定的后缀结尾。

方法二：使用glob模块

除了使用os模块，我们还可以使用Python的glob模块来寻找指定后缀的文件。glob模块提供了一个简洁的方式来匹配文件路径，支持通配符的使用。下面是示例代码：

import glob

def find_files_with_extension(directory, extension):
    pattern = os.path.join(directory, f'*{extension}')
    files = glob.glob(pattern, recursive=True)
    return files

directory = '/path/to/directory'
extension = '.txt'
files = find_files_with_extension(directory, extension)
print(files)

在这个示例中，我们使用glob.glob函数来匹配指定目录下以指定后缀结尾的文件。我们使用os.path.join函数来构建一个匹配模式，*表示通配符，代表任意字符序列。通过设置recursive=True参数，我们可以递归地搜索子目录。

方法三：使用Path对象

Python 3.4及以上版本提供了一个更加现代化的文件操作方式，即使用Path对象。Path对象提供了一组方便的方法来操作文件路径，包括寻找指定后缀的文件。下面是示例代码：

from pathlib import Path

def find_files_with_extension(directory, extension):
    path = Path(directory)
    files = list(path.glob(f'**/*{extension}'))
    return [str(file) for file in files]

directory = '/path/to/directory'
extension = '.txt'
files = find_files_with_extension(directory, extension)
print(files)

在这个示例中，我们使用Path对象的glob方法来匹配指定目录下以指定后缀结尾的文件。我们使用**通配符来递归地搜索子目录，并使用*通配符来匹配任意字符序列。最后，我们将Path对象转换为字符串形式的文件路径列表。

总结

本文介绍了三种常用的方法来寻找指定后缀的文件：使用os模块和字符串切片、使用glob模块以及使用Path对象。这些方法各有优缺点，可以根据具体需求选择合适的方法。通过掌握这些方法，我们可以更加方便地进行文件操作和处理，提高编程效率。

以下是示例代码中的甘特图：

gantt
    dateFormat  YYYY-MM-DD
    title Python寻找指定后缀文件

    section 方法一
    使用os模块和字符串切片: 2022-01-01, 3d

    section 方法二
    使用glob模块: 2022-01-04, 3d

    section 方法三
    使用Path对象: 2022-01-07, 3d