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