python – 如何缩短多个IF … IN … OR语句？

发布时间：2020-12-20 13:00:11 所属栏目：Python 来源：网络整理

导读：参见英文答案 Checking file extension????????????????????????????????????10个如何缩短以下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 我在思考 f

参见英文答案 > Checking file extension????????????????????????????????????10个
如何缩短以下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','e.JPG']
images = [x for x in files if ('.jpg' or '.png' or '.JPG') in x]
print images

这不起作用.

与这篇文章相反：Checking file extension,我也对一个概括感兴趣,它不关注文件的结尾.

解决方法

这有点短

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

这是有效的,因为endswith()可以使用元组输入,因为你可以看到in the docs.

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

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!