Python - 列出目录中的所有CSV文件

在Python中，我们经常需要处理CSV文件。而当我们需要读取一个目录中的所有CSV文件时，我们可以使用以下方法来实现。

方法一：使用os库和文件扩展名来过滤文件

import os

def list_csv_files(directory):
    csv_files = []
    for file in os.listdir(directory):
        if file.endswith(".csv"):
            csv_files.append(file)
    return csv_files

这个方法遍历给定目录中的所有文件，然后使用endswith()方法来检查文件扩展名是否为.csv。如果是，则将文件名添加到csv_files列表中。最后，返回包含CSV文件名的列表。

使用方法：

csv_directory = "path/to/directory"
csv_files = list_csv_files(csv_directory)
print(csv_files)

方法二：使用glob模块来匹配文件路径

Glob模块提供了一个简便的方法可以根据通配符模式来匹配文件路径。

import glob

def list_csv_files(directory):
    csv_files = glob.glob(directory + "/*.csv")
    return csv_files

这个方法使用glob.glob()函数来匹配目录中所有的*.csv文件。返回的是一个包含匹配文件路径的列表。

使用方法：

csv_directory = "path/to/directory"
csv_files = list_csv_files(csv_directory)
print(csv_files)

无论是使用方法一还是方法二都可以列出给定目录中的所有CSV文件。你可以根据自己的需要选择其中一种方法来使用。

希望这个介绍对你有所帮助！