如何将 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)