How to find char in string and get all the indexes?

Question

I got some simple code:

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

The function only return the first index. If I change return to print, it will print 0 0 0. Why is this and is there any way to get 0 1 2?

Answer 1

This is because str.index(ch) will return the index where ch occurs the first time. Try:

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

This will return a list of all indexes you need.

P.S. Hugh's answer shows a generator function (it makes a difference if the list of indexes can get large). This function can also be adjusted by changing [] to ().