find index of values containing keyword in array python

Question

I want to find the index of values that contain a keyword in an array.

For example:

A = ['a1','b1','a324']
keyword = 'a'

I want to get [0,2], which is the index of a1, a324

I tried this list(filter(lambda x:'a' in x, A)) But get ['a1','a324'] rather than the index.

Answer 1

Use enumerate with a list-comprehension:

A = ['a1','b1','a324']
keyword = 'a'

print([i for i, x in enumerate(A) if keyword in x])
# [0, 2]

Answer 2

Simply write:

A = ['a1','b1','a324']
keyword = 'a'
indices = [i for i in range(len(A)) if keyword in A[i]]
print(indices)