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])