Python 统计具有特定扩展名的文件数量

在本文中，我们将介绍如何使用Python统计具有特定扩展名的文件数量。我们将使用Python的内置库来实现这个功能。

方式一：使用os库

我们可以使用Python的os库来遍历文件夹并计算指定扩展名的文件数量。下面的示例代码演示了如何统计目标文件夹中以”.txt”为扩展名的文件数量。

import os

def count_files_with_extension(folder_path, extension):
    count = 0

    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith(extension):
                count += 1

    return count

folder = "/path/to/folder"
extension = ".txt"

file_count = count_files_with_extension(folder, extension)
print(f"The number of {extension} files in {folder} is {file_count}.")

在上面的代码中，我们定义了一个count_files_with_extension函数，该函数接受文件夹路径和目标扩展名作为参数。使用os.walk函数遍历了目标文件夹及其子文件夹中的所有文件。通过判断文件是否以目标扩展名结尾，来计算符合条件的文件数量。

方式二：使用glob库

除了使用os库之外，我们还可以使用Python的glob库来实现同样的功能。glob库提供了一个简单的方法来查找匹配特定模式的文件。下面的示例代码演示了如何使用glob库统计目标文件夹中以”.txt”为扩展名的文件数量。

import glob

def count_files_with_extension(folder_path, extension):
    pattern = f"{folder_path}/**/*{extension}"
    file_list = glob.glob(pattern, recursive=True)

    return len(file_list)

folder = "/path/to/folder"
extension = ".txt"

file_count = count_files_with_extension(folder, extension)
print(f"The number of {extension} files in {folder} is {file_count}.")

在上面的代码中，我们定义了一个count_files_with_extension函数，该函数接受文件夹路径和目标扩展名作为参数。使用glob库的glob函数配合通配符**来匹配所有符合条件的文件。最后返回文件列表的长度，即统计的文件数量。

无论是使用os库还是glob库，我们都可以根据需要自定义文件夹路径和扩展名，方便地统计目标文件夹中具有特定扩展名的文件数量。