如何仅返回具有特定文件类型的文件？

Question

这是我的代码：

import os

def read_file(path):
    name_file = os.listdir(path)
    return name_file

path = "/home/ich/Schreibtisch/fotki"
print(read_file(path)) 

这仅返回列表中文件夹中的所有文件。但我希望他们以".jpg"结尾。文件夹中有很多文件（.jpg，.txt，...），但我只希望这些.jpg文件。

Answer 1

使用os.listdir和.endswith

import os

path = "/home/ich/Schreibtisch/fotki"

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