mercredi 21 décembre 2016

How to shorten multiple IF .... IN ... OR statements?

How can I shorten the following 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

I was thinking in terms of

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

which does not work.

Aucun commentaire:

Enregistrer un commentaire