Python字符串中查找子串小技巧

惭愧啊，今天写了个查找子串的Python程序被BS了…

如果让你写一个程序检查字符串s2中是不是包含有s1。也许你会很直观的写下下面的代码：

#determine whether s1 is a substring of s2 

def isSubstring1(s1,s2): 

    tag = False 

    len1 = len(s1) 

    len2 = len(s2) 

    for i in range(0,len2): 

        if s2[i] == s1[0]: 

            for j in range(0,len1): 

                if s2[i]==s1[j]: 

                    tag = True 

    return tag

可是这是Python，我们可以利用字符串自带的find()方法，于是可以这样：

def isSubstring2(s1,s2): 

    tag = False 

    if s2.find(s1) != -1: 

        tag = True 

    return tag

悲情的事就在于此，原来Python中的关键字"in”不仅可以用于列表、元祖等数据类型，还可以用于字符串。所以，这里只需要直接一行代码搞定：

def isSubstring3(s1,s2):

    return s1 in s2

后知后觉了，惭愧;-)

类似的，假设要在字符串中，查找多个子串是否存在，并打印出这些串和首次出现的位置：

def findSubstrings(substrings,destString):

    res =  map(lambda x:str([destString.index(x),x]),filter(lambda x:x in destString,substrings))

    if res:

        return ', '.join(list(res))

 

;-)  very cool~

UPDATE: 如果你不习惯最后面这种看起来很复杂的语法也没关系，可以使用列表解析，更加简洁：

def findSubstrings(substrings,destString):

    return ', '.join([str([destString.index(x),x]) for x in substrings if x in destString])