我的 python 代码在目录“MAIN”中。 在 MAIN 中,我有一个文件夹“DATA”。
在 DATA 中,我有一个 CSV 文件和另一个文件夹“_OLD”
我想将 DATA 文件夹中的 CSV 导入到 pandas 数据框中,但我不确定它的名称。
我该怎么做?
注意:DATA 文件夹中始终只有一个 CSV 和 _OLD 文件夹。因此,我需要一种方法来访问该文件夹中的唯一文件。
请您参考如下方法:
你可以这样做:
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
