我有两个排序的数组h
和M
:
h = np.array(['blue', 'red', 'white'])
M = np.array(['blue', 'green', 'orange', 'red', 'white'])
并想找到每个元素h
出现的索引M
我np.where
可以为此吗?
此外,也可能是 的元素h
可能不会出现M
在这种情况下,我不需要它的索引。元素不重复。
这是另一种方式,是一个单行,将返回ndarray
匹配项的索引。
np.nonzero(np.in1d(M, h))[0]
输出
array([0, 3, 4])
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句