如何将os.listdir输出保存为列表

Question

我试图将os.listdir("./oldcsv")的所有条目分别保存在一个列表中，但是我不知道如何在处理输出之前对其进行操作。

我想做的是生成一个包含文件夹中所有* .csv文件的绝对路径名的列表，以后可用于轻松地操纵这些文件的内容。我不想在脚本中添加很多硬编码的路径名，因为它很烦人且难以阅读。

import os

for file in os.listdir("./oldcsv"):
    if file.endswith(".csv"):
        print(os.path.join("/oldcsv", file))

通常我会在.append处使用循环，但是在这种情况下我不能这样做，因为os.listdir似乎只是在创建内容的“斑点”。也许那里有一个简单的解决方案，但是我的大脑却没有想到。

Answer 1

标准库中有一个glob模块，可以通过一个函数调用解决您的问题：

import glob 

csv_files = glob.glob("./*.csv")  # get all .csv files from the working dir
assert isinstance(csv_files, list)