有没有可能把功能减少到一行？

Question

我写了一个图像检查器，现在我想知道如何将这个函数中的行数减少到一行（如果可能的话）。你知道吗

myFiles = ['image94.jpg','image95.png','image96.jpg','movie97.mov']
suff = ('.jpg', '.png')

# Check if files are images
def checker(suff):
    imageFiles = []
    for files in myFiles:
        if files.endswith(suff):
            imageFiles.append(files)
    return imageFiles

if checker(suff): print checker(suff)

Answer 1

使用条件列表：

[filename for filename in myFiles if filename.endswith(suff)]

Answer 2

试试兰博达斯的魔力。。你知道吗

f = lambda suff,l:filter(lambda x: '.'+x.split('.')[-1] in suff,l)

用它来称呼它

f(suff,myFiles)