[Python知识库]Python百练成钢004-查找字符串中所有子串索引

这是【Python 百练成钢】系列文章的第 004 篇，计划完成 100 道练习题。

本文环境：python3.8

需求描述：
有两个字符串 s1, s2，找出 s1 中包含的 s2 的起始索引，若有多个，则返回多个索引。

def get_all_index(fullstr: str, substr: str):
    """
    返回 substr 在 fullstr 中的起始索引，若匹配多个，则返回多个索引
    :param fullstr: 大串
    :param substr: 子串
    :return: 索引
    """
    res_index = []
    i = -1
    while substr in fullstr:
        # find 返回第一次匹配的索引
        i = fullstr.find(substr)
        if not res_index:
            res_index.append(i)
        else:
            res_index.append(i+res_index[-1]+len(substr))
        # 截取出剩余子串，继续匹配
        fullstr = fullstr[i+len(substr):]

    return f"匹配到的子串索引为：{res_index}"


print(get_all_index('aabcdefabaaa', 'ab'))

输出结果为：

匹配到的子串索引为：[1, 7]