从列表中删除不属于某个文件路径的文件 - Python

Question

作为我项目的一部分，我的工具要求用户输入一个文件夹，该文件夹中包含的所有文件都存储在一个列表中，供以后使用。我需要这样做，以便列表只包含“.jpg”或“.tiff”文件格式的文件，因此删除不是该文件格式的任何文件。我似乎无法弄清楚如何做到这一点。如何从列表中删除任何其他文件格式？我将在下面发布一个我已经尝试过的示例。

files = os.listdir(file_dir)

for file in files:
    if not files.endswith(".jpg", ".tiff"):
        files = files.remove(file)

Answer 1

用

files = [f for f in os.listdir(file_dir) if (f.endswith(".jpg") or f.endswith(".tiff"))]