作为我项目的一部分,我的工具要求用户输入一个文件夹,该文件夹中包含的所有文件都存储在一个列表中,供以后使用。我需要这样做,以便列表只包含“.jpg”或“.tiff”文件格式的文件,因此删除不是该文件格式的任何文件。我似乎无法弄清楚如何做到这一点。如何从列表中删除任何其他文件格式?我将在下面发布一个我已经尝试过的示例。
files = os.listdir(file_dir)
for file in files:
if not files.endswith(".jpg", ".tiff"):
files = files.remove(file)
Run Code Online (Sandbox Code Playgroud)