python - 如何检查数组A中的元素是否存在于数组B中并返回B的索引

Question

我花了时间搜索这个问题，但找不到任何现有的解决方案。假设我有数组 A 和数组 B

A = np.array([2,3,3,3,4,5])
B = np.array([3,4])

如果我使用np.isin: 它会返回array([False, True, True, False])。但是，我真正想要的是一些可以返回的功能

array([-1,  0, 0, 0, 1, -1])

其中第一个和最后一个-1表示在B中找不到A中的元素，0表示在B中的索引0中找到元素，1为B中的索引1。

任何优雅的解决方案？

标签： pythonpython-3.xnumpy