python - 文件名未知时在python中访问CSV

Question

我的 python 代码在目录“MAIN”中。在 MAIN 中，我有一个文件夹“DATA”。

在 DATA 中，我有一个 CSV 文件和另一个文件夹“_OLD”

我想将 DATA 文件夹中的 CSV 导入到 pandas 数据框中，但我不确定它的名称。

我该怎么做？

注意:DATA 文件夹中始终只有一个 CSV 和 _OLD 文件夹。因此，我需要一种方法来访问该文件夹中的唯一文件。

Answer 1

你可以这样做:

from glob import glob
filename = glob("MAIN/DATA/*.csv")[0]

编辑:另一种方法是使用pathlib 库。一个区别是 Path 对象的 glob() 方法返回给定目录中的文件生成器。发出的对象是 Path 对象(如果需要，可以使用 str 字符串构造函数将其转换为字符串。

from pathlib import Path

for filename in Path("MAIN/DATA").glob("*.csv"):
    # do something