python - 如何在字符串中查找char并获取所有索引？

Question

我得到了一些简单的代码:

def find(str, ch):
    for ltr in str:
        if ltr == ch:
            return str.index(ltr)
find("ooottat", "o")

该函数只返回第一个索引。如果我将返回更改为打印，它将打印 0 0 0。为什么会这样，有什么方法可以得到 0 1 2？

Answer 1

这是因为 str.index(ch) 将返回 ch 第一次出现的索引。试试:

def find(s, ch):
    return [i for i, ltr in enumerate(s) if ltr == ch]

这将返回您需要的所有索引的列表。

附: Hugh 的回答显示了一个生成器函数(如果索引列表变大，它会有所不同)。这个函数也可以通过将[]改为()来调整。