在numpy数组中查找多个元素的索引

Question

给定一个整数数组，我需要找到存储在不同数组中的许多元素的索引。这是：

import numpy as np
a1 = np.array([ 4, 5, 6, 1, 2, 3, 7, 86, 9, 15])
a2 = np.array([ 2, 3, 5, 6, 9])

a1是我的初始元素数组，而a2是包含我需要a1中索引的元素的数组。

在这种情况下，结果应为：

a3 = ([4, 5, 1, 2, 8])

这似乎是一个相当简单的操作，但我无法弄清楚如何去做。

Answer 1

你可以试试这个：

In [378]: (a1[:, None] == a2).argmax(axis=0)
Out[378]: array([4, 5, 1, 2, 8], dtype=int64)