Python小技巧 - 子串查詢

weixin_34015860發表於2019-01-07

慚愧啊,今天寫了個查詢子串的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])
 

 

 

相關文章