这个问题在这里已有答案:
我想找到包含数组中关键字的值的索引。
例如:
A = ['a1','b1','a324']
keyword = 'a'
我想得到[0,2]
,这是a1, a324
的索引
我试过这个list(filter(lambda x:'a' in x, A))
但是得到['a1','a324']
而不是索引。
使用enumerate
列表理解:
A = ['a1','b1','a324']
keyword = 'a'
print([i for i, x in enumerate(A) if keyword in x])
# [0, 2]
只需写:
A = ['a1','b1','a324']
keyword = 'a'
indices = [i for i in range(len(A)) if keyword in A[i]]
print(indices)