如何缩短以下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'))]总结
以上是内存溢出为你收集整理的python – 如何缩短多个IF … IN … OR语句?全部内容,希望文章能够帮你解决python – 如何缩短多个IF … IN … OR语句?所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)