Python更优雅的方式来编写这个IF OR语句？

Question

是否有更优雅的方式来写这个陈述？我发现自己写的相似很多.

if ('.jpg' in imagename[0].lower() or '.png' in imagename[0].lower() or '.jpeg' 
    in imagename[0].lower()):

Answer 1

你可能想要

if imagename[0].lower().endswith((".jpg", ".jpeg", ".png")):
    # whatever

请注意,str.endswith()接受字符串元组.另请注意,此代码与您的代码不同,因为它只检查文件名末尾提到的字符串,但可能是您的意图.