在字符串列表中查找子字符串的所有实例并返回找到子字符串的字符串的索引

Question

我如何在字符串列表中找到子字符串的所有实例并返回在 Python 中找到子字符串的字符串的索引？

例如：

sub_string = "tree"
my_list = ["banana", "tree", "trees", "street"]

所需的输出是： [1,2,3]因为在索引为[1,2,3]的字符串中找到树。

我有一个函数的形式，但它只返回子字符串索引的第一个实例，并且不能识别字符串中的子字符串（例如街道中的树）。

def inside_search(a_list, search_term):
    if search_term in a_list:
        index = a_list.index(search_term, 0, -1)
        return [index]
    else:
        return []

cats_and_dogs_list = ["cat", "cats", "dog", "dogs", "catsup"]
print(inside_search(cats_and_dogs_list, "cat"))

我的函数返回[0]但我希望它返回[0,1,4]

我已经尝试使用几种方法解决了这个问题，但除了[0]之外我似乎什么也不能返回。

Answer 1

将列表理解与enumerate ：

>>> [i for i, w in enumerate(my_list) if sub_string in w]
[1, 2, 3]

如果你想使用一个函数：

def inside_search(a_list, search_term):
    result = list()
    for i, word in enumerate(a_list):
        if search_term in word:
            result.append(i)
    return result

>>> inside_search(cats_and_dogs_list, "cat")
[0, 1, 4]

Answer 2

这就是你要找的：

print([my_list.index(item) for item in my_list if sub_string in item])

希望这有帮助:) 干杯！