经常处理字符串,想检查字符串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