📜  numpy 查找匹配值的索引 - Python 代码示例

📅  最后修改于: 2022-03-11 14:46:08.324000             🧑  作者: Mango

代码示例1
import numpy as np
# init arrays
a = np.array([1,2,3,2,3,4,3,4,5,6])
b = np.array([7,2,10,2,7,4,9,4,9,8])

#using enumerate, list comprehension and set
print([key for key, val in enumerate(a) if val in set(b)])
# output
# [1, 3, 5, 7]