python基础学习 | 我有个数组a，1个数组b，我想得到数组b元素在数组a中的index

问题描述

我有个数组a，1个数组b，我想得到数组b元素在数组a中的index

示例代码

import numpy as np

a = np.array([10, 20, 30, 40, 50])  # 数组a
b = np.array([20, 40])    # 数组b

indices = np.where(np.isin(a, b))[0]
print(indices)

# 输出结果
[1 3]