Я хочу найти индекс значений, содержащий ключевое слово в массиве.
Например:
A = ['a1','b1','a324']
keyword = 'a'
Я хочу получить [0,2]
, который является индексом a1, a324
Я попробовал этот list(filter(lambda x:'a' in x, A))
Но вместо [index ['a1','a324']
вместо [index ['a1','a324']
.
Просто напишите:
A = ['a1','b1','a324']
keyword = 'a'
indices = [i for i in range(len(A)) if keyword in A[i]]
print(indices)
Используйте enumerate
со списком:
A = ['a1','b1','a324']
keyword = 'a'
print([i for i, x in enumerate(A) if keyword in x])
# [0, 2]