複数の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 

の観点で考えていました。

この投稿とは対照的に：Checking file extension、私はまた、ファイルの終わりに焦点を当てていない一般化に興味があります。

Answer 1

これは、あなたがin the docsを見ることができるようendswith()は、入力のためのタプルを取ることができますので、それが動作します

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 

少し短いです。

あなたも、このような何かがそれを行う必要があり、それ大文字小文字を区別しない

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

Answer 2

作るためにこれを行うことができます。

import os 

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

Answer 3

方法について：

files = ['a.txt', 'b.jpg', 'c.png', 'd.JPG', 'e.JPG'] 
formats = ('.jpg', '.png', '.JPG') 

# this gets you the images 
images = [file for file in files if any (format in file for format in formats)) 

# The above is equivalent to the following statement which is longer 
# and looks complicated but probably easy to understand for someone new to [python list comprehension][1] 
images = [file for file in files if any (format for format in formats if format in file)) 

しかし、しかし、有しますを確認する場合は、実際にthis answerを使用する必要があります10。私はただあなたの前提を広げました（あなたの質問に基づいて、inを使用しました）。リスト内包上

推奨読書は：python documentation