如何缩短多个IF .... IN ... OR语句？

Question

如何缩短以下MWE？

files = ['a.txt', 'b.jpg', 'c.png', 'd.JPG', 'e.JPG']
images = [x for x in files if '.jpg' in x or '.png' in x or '.JPG' in x]
print images

我在思考

files = ['a.txt', 'b.jpg', 'c.png', 'd.JPG', 'e.JPG']
images = [x for x in files if ('.jpg' or '.png' or '.JPG') in x]
print images

这不起作用.

与这篇文章相反:检查文件扩展名,我也对一个概括感兴趣,它不关注文件结尾.

Answer 1

这有点短

files = ['a.txt', 'b.jpg', 'c.png', 'd.JPG', 'e.JPG']
images = [x for x in files if x.endswith(('.jpg','.png','.JPG'))]
print images

它的工作原理是因为endswith()你可以在文档中看到输入的元组.

你甚至可以这样做,使它不区分大小写

images = [x for x in files if x.lower().endswith(('.jpg','.png'))]