菜鸟笔记
提升您的技术认知

python检查字符串中的结束标记-ag真人游戏

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

 

网站地图