numpy为多维数组获取数字的多个索引

Question

给定两个数组：

array_1 =  np.array([[1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15]])

及

array_2 =  np.array([[2,3,5], [6,8,9], [12,13,14]])

我想要数组_1上数组_2的值的基于行的索引，akas

np.array([[1,2,4], [0,2,3], [1,2,3]])

作为期望的输出

我应该如何处理numpy函数

Answer 1

像这样的东西？：

import numpy as np

array_1 =  np.array([[1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15]])
array_2 =  np.array([[2,3,5], [6,8,9], [12,13,14]])

result = {}

for index, list_val in np.ndenumerate(array_2):
    print(f'Finding: {list_val} in {array_1[index[0]]}')
    print()
    index_in_array_1 = np.where(array_1[index[0]] == list_val )
    print(f'match at : {index_in_array_1}')
    print()

结果：