Python 统计文件数

引言

在日常的文件管理中，我们经常需要统计某个文件夹中的文件数量。Python作为一门强大的编程语言，提供了丰富的库和函数来实现这个功能。本文将介绍如何使用Python统计文件数，并提供相应的代码示例。

问题描述

我们需要统计一个文件夹中的文件数量，包括文件夹下的子文件夹中的文件。具体而言，问题可以定义为：给定一个文件夹路径，如何统计该文件夹中的文件数量？

解决方案

1. 使用os模块

Python的os模块提供了许多与操作系统交互的函数，包括文件和目录的操作。我们可以利用os模块中的函数来实现文件统计功能。

首先，我们需要导入os模块：

import os

然后，我们可以使用os模块中的walk函数来遍历文件夹及其子文件夹中的文件。walk函数会返回一个生成器，每次迭代时返回当前文件夹路径、子文件夹列表和文件列表。我们可以使用一个循环来遍历所有文件夹，并统计文件数量。

下面是一个使用os模块统计文件数的示例代码：

import os

def count_files(folder_path):
    file_count = 0
    for root, dirs, files in os.walk(folder_path):
        file_count += len(files)
    return file_count

# 统计当前文件夹中的文件数量
file_count = count_files('.')
print("当前文件夹中的文件数量：", file_count)

在上述代码中，count_files函数接受一个文件夹路径作为参数，返回该文件夹中的文件数量。使用os.walk函数遍历文件夹及其子文件夹，files列表保存了当前文件夹中的所有文件，通过取files列表的长度即可得到文件数量。

2. 使用glob模块

除了os模块，Python还提供了glob模块来查找文件路径匹配的文件。glob模块提供了类似于Unix中的通配符功能，可以方便地查找符合特定条件的文件。

首先，我们需要导入glob模块：

import glob

然后，可以使用glob.glob函数来获取匹配特定条件的文件列表。我们可以将该函数的参数设为文件夹路径加上/**/*，表示匹配该文件夹及其子文件夹中的所有文件。

下面是一个使用glob模块统计文件数的示例代码：

import glob

def count_files(folder_path):
    file_list = glob.glob(folder_path + '/**/*', recursive=True)
    return len(file_list)

# 统计当前文件夹中的文件数量
file_count = count_files('.')
print("当前文件夹中的文件数量：", file_count)

在上述代码中，count_files函数接受一个文件夹路径作为参数，返回该文件夹中的文件数量。使用glob.glob函数获取所有匹配的文件路径列表，通过取列表的长度即可得到文件数量。需要注意的是，参数recursive=True表示递归地搜索子文件夹。

总结

本文介绍了两种使用Python统计文件数的方法：使用os模块和glob模块。os模块提供了与操作系统交互的函数，可以遍历文件夹及其子文件夹，统计文件数量。glob模块提供了通配符功能，可以方便地查找符合特定条件的文件。

下表对比了使用os模块和glob模块的优缺点：

方法 优点 缺点
os模块 简单易用，无需额外安装任何库 遍历文件夹及其子文件夹比较耗时，不适用于大型文件夹
glob模块 方便快捷，通过通配符支持更多的文件匹配条件 需要额外安装glob库

根据实际需求选择合适的方法来统计文件数。

参考资料