查找字符串中字符的所有位置

我正在尝试使用非常基本的技能集查找 python 字符串中字符的所有索引号。例如，如果我有字符串“Apples are Excellent”，并且我想找到字符串中“a”所在的位置。 我理想的输出是：

0
7
14
19

这些是字符串中出现“a”的所有位置（我认为）

这是我到目前为止的代码：

sentence = input("Input a string: ")
for ch in sentence:
    x = sentence.find('o')


print(x)

在这里我正在寻找“o”而不是“a”。我的思考过程是，对于字符串中的每个字符，查找函数将返回“o”的位置。因为我不知道输入字符串一定有多长，所以我使用了 for 循环。我能够找到并打印出“o”的第一个实例，但不是全部。我应该怎么办？提前致谢！

Using enumerate是标准的方法。虽然，您可以利用速度str.find用于时间关键的操作。

Code

def find_all(s, c):
    idx = s.find(c)
    while idx != -1:
        yield idx
        idx = s.find(c, idx + 1)

print(*find_all('Apples are totally awesome', 'o')) # 12 23

为了优雅并考虑非常大的字符串，我将上面的返回生成器。当然可以把它投射到list如果需要的话。

基准

这是针对使用解决方案的基准enumerate和列表理解。两种解决方案都具有线性时间复杂度，但是str.find明显更快。

import timeit

def find_all_enumerate(s, c):
    return [i for i, x in enumerate(s) if c == x]

print(
    'find_all:',
    timeit.timeit("list(find_all('Apples are totally awesome', 'o'))",
                  setup="from __main__ import find_all")
)

print(
    'find_all_enumerate:',
    timeit.timeit("find_all_enumerate('Apples are totally awesome', 'o')",
                  setup="from __main__ import find_all_enumerate")
)

Output

find_all: 1.1554179692960915
find_all_enumerate: 1.9171753468076869