最近遇到一个问题,要求查找一个短字符串在长字符中的位置,发现Python没有类似数据库中instr的方法,于是自己试着写了一个:
#整体思路是遍历长字符串,匹配短字符串第一个字符p[0]
t = "AVERDXIVYERDIANRDXIXYAXDRDXI" #长字符串
p = "RDXI" #短字符串
#字典,遍历长字符串,匹配短字符串第一个字符p[0]
#如果匹配到,key存放下标,所有匹配的下标都记录
#存匹配位置起,截取和短字符串长度相同的子串,存于value
d1 = {}
#如果value和短字符串内容相同,把对应的key添加到列表中
key_list = []
def str_instr(long_str, shortstr, n):
#遍历长字符串的每个字符,匹配短字符串第一个字符p[0]
for i in range(0,len(t)):
#每次匹配到,就记录一次下标,同时向右截取短字符串长度的子串
if t[i] == p[0]:
#记录长字符中对应的下标,存于字典d1作为key
#从当前位置开始向右截取和短字符串长度相同的子串,存于字典作为value
d1[i] = t[i:i+4:1]
#字典d1记录了所有和短字符串首字符相同的子串
#遍历字典,如果value和短字符串相同,就把key提取出来,存在列表中
for k,v in d1.items():
if v == p:
key_list.append(k)
#列表中的最小值,即为p在t中第一次出现的下标
#min_index = min(key_list)
#return min_index
#第n次出现子字符串的下标:
#列表的下标从0开始,因此要减1
key_list.sort()
n_index = key_list[n-1]
return n_index
#测试方法
str_instr(t,p,1) #3
str_instr(t,p,2) #15