在numpy中，如何快速找到一个1Dndarray `b`在另一个1Dndarray `a`中的元素的索引？

在numpy中，可以使用函数numpy.where()来快速找到一个1D ndarray b在另一个1D ndarray a中的元素的索引。

具体步骤如下：

1. 导入numpy库：import numpy as np
2. 使用numpy.where()函数来查找元素的索引：indices = np.where(a == b)
   • a == b表示对数组a进行逐元素比较，返回一个布尔型数组，表示a中的元素是否与b相等。
   • np.where()函数会返回满足条件的元素的索引，以元组的形式返回，其中第一个元素是满足条件的元素的行索引，第二个元素是满足条件的元素的列索引（对于1D数组来说，列索引始终为0）。

• 获取索引值：indices[0]
  • indices[0]表示获取满足条件的元素的行索引，即在数组a中与数组b相等的元素的索引。

下面是一个完整的示例代码：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([3, 5])

indices = np.where(a == b)
result = indices[0]

print(result)

输出结果为：

[2 4]

这表示在数组a中，元素3的索引为2，元素5的索引为4。