python 获取路径下所有的csv文件

Python获取路径下所有的CSV文件

在日常的数据处理工作中，经常会遇到需要处理多个CSV文件的情况。如果手动一个一个去找并读取这些文件，不仅费时费力，还容易出错。幸运的是，Python提供了许多库和方法来处理文件，其中包括获取指定路径下所有的CSV文件。本文将介绍如何使用Python来获取指定路径下所有的CSV文件，并给出相应的代码示例。

1. 获取指定路径下的所有文件

要获取指定路径下的所有文件，可以使用os模块中的walk函数。这个函数会递归地遍历指定路径下的所有文件夹和文件，并返回一个生成器对象。我们可以通过遍历这个生成器对象，获取到所有文件的路径。

下面是使用os.walk函数获取指定路径下所有文件的代码示例：

import os

def get_all_files(path):
    # 使用os.walk函数递归遍历指定路径下的所有文件夹和文件
    for root, dirs, files in os.walk(path):
        for file in files:
            yield os.path.join(root, file)

在这个代码中，path参数是指定路径的字符串。os.walk函数的返回值是一个生成器对象，我们通过yield关键字将每个文件的路径逐个返回。

2. 筛选出所有的CSV文件

在获取到所有文件的路径后，我们需要筛选出其中的CSV文件。我们可以使用splitext函数来获取文件的扩展名，然后判断是否为CSV文件。

下面是筛选出所有CSV文件的代码示例：

def get_csv_files(path):
    for file in get_all_files(path):
        # 使用os.path.splitext函数获取文件的扩展名
        _, ext = os.path.splitext(file)
        if ext == '.csv':
            yield file

在这个代码中，我们通过os.path.splitext函数获取到文件的扩展名。然后判断扩展名是否为.csv，如果是则将文件的路径返回。

3. 打印获取到的CSV文件路径

使用上述的代码，我们已经可以获取到指定路径下所有的CSV文件路径了。为了验证代码的正确性，我们可以将获取到的文件路径打印出来。

下面是打印获取到的CSV文件路径的代码示例：

if __name__ == '__main__':
    path = '/path/to/csv/files'  # 替换成指定路径
    for csv_file in get_csv_files(path):
        print(csv_file)

在这个代码中，我们使用for循环遍历获取到的CSV文件路径，并使用print函数将路径打印出来。

4. 示例应用：统计CSV文件数量

上面的代码展示了如何获取指定路径下所有的CSV文件。接下来，我们将给出一个示例应用：统计CSV文件的数量。

下面是统计CSV文件数量的代码示例：

if __name__ == '__main__':
    path = '/path/to/csv/files'  # 替换成指定路径
    count = 0
    for csv_file in get_csv_files(path):
        count += 1
    print(f'Total number of CSV files: {count}')

在这个代码中，我们定义了一个计数器count，在每次遍历到CSV文件时，将计数器加1。最后使用print函数将CSV文件数量打印出来。

总结

本文介绍了如何使用Python获取指定路径下所有的CSV文件。我们使用os.walk函数遍历指定路径下的所有文件夹和文件，然后使用os.path.splitext函数筛选出CSV文件。最后，我们给出了一个示例应用，统计CSV文件的数量。

使用Python获取路径下所有的CSV文件可以极大地提高工作效率，减少手动查找和处理文件的时间。希望本文对你有所帮助！

状态图

下面是一个简单的状态图，表示获取路径下所有CSV文件的过程：

stateDiagram
    [*] --> 获取所有文件路径
    获取所有文件路径 --> 筛选CSV文件
    筛选CSV文件 --> 打印文件路径
    打印文件路径 --> [*]

参考链接

• [Python