Python 使用文件扩展名通配符在os.listdir(path)中

在本文中，我们将介绍如何使用Python的os.listdir()函数来使用文件扩展名通配符。os.listdir()函数用于列出指定路径下的所有文件和文件夹。我们经常需要筛选具有特定扩展名的文件，例如只获取所有的.jpg图片文件。使用文件扩展名通配符可以解决这个问题。

阅读更多：Python 教程

文件扩展名通配符

文件扩展名通配符是一种用于匹配一类文件的字符串模式。在Python中，我们可以使用通配符字符”*”和”?”来表示任意长度和单个字符的通配符。在文件名中，我们可以使用通配符来匹配具有特定扩展名的文件。

下面是通配符的示例：

“*”：匹配任意长度的任意字符
“?”：匹配任意单个字符

使用通配符获取指定扩展名的文件

要使用通配符获取指定扩展名的文件，我们可以使用os.listdir()函数列出指定路径下的所有文件和文件夹，然后使用字符串函数进行筛选。以下是一个示例：

import os

def get_files_with_extension(path, extension):
    files = os.listdir(path)
    filtered_files = [file for file in files if file.endswith(extension)]
    return filtered_files

path = "C:/Users/username/Documents"
extension = ".jpg"
jpg_files = get_files_with_extension(path, extension)
print(jpg_files)

上述示例中，我们定义了一个名为get_files_with_extension()的函数，它接受一个路径和一个扩展名作为参数。在函数内部，我们使用os.listdir()函数列出路径下的所有文件和文件夹，并使用列表解析筛选出以指定扩展名结尾的文件。最后，我们返回这些文件并打印输出。

支持通配符的扩展名匹配

除了使用固定的扩展名进行匹配外，我们还可以使用通配符来进行模糊匹配。以下是一个示例：

import os
import fnmatch

def get_files_with_wildcard(path, wildcard):
    files = os.listdir(path)
    filtered_files = [file for file in files if fnmatch.fnmatch(file, wildcard)]
    return filtered_files

path = "C:/Users/username/Documents"
wildcard = "*.txt"
txt_files = get_files_with_wildcard(path, wildcard)
print(txt_files)

上述示例中，我们导入了fnmatch模块，该模块提供了支持通配符的文件名匹配功能。我们定义了一个名为get_files_with_wildcard()的函数，它接受一个路径和一个通配符作为参数。在函数内部，我们使用os.listdir()函数列出路径下的所有文件和文件夹，并使用列表解析筛选出与通配符匹配的文件。最后，我们返回这些文件并打印输出。