Python检查字符串中的结束标记

经常处理字符串，想检查字符串s的结束标志，如果结束标志只有一个还好办，s.endswith(end1)就可以了，如果有多个就需要使用s.endswith(end1), s.endswith(end2), s.endswith(end3)这种笨重的方法。对于这种常见场景的处理，使用itertools模块中的imap函数可以很好解决这个问题。

一个经典的应用是打印出当前目录中所有的图片文件:

import os
import itertools

def endsWith(s, lst):
    def anyTrue(func, seq):
        return True in itertools.imap(func, seq)
    return anyTrue(s.endswith, lst)

if __name__ == '__main__':
    end_tag_list = ['.jpg', '.png', '.gif']
    for filename in os.listdir('.'):
        if endsWith(filename, end_tag_list):
            print(filename)

就可以输出所有符合条件

c2.jpg
c3.jpg
c4.jpg
c5.jpg
c6.jpg
client1.png
client2.png
client3.png
client4.png
client5.png
client6.png
clouds.gif
codebg.gif