numpy 多维查找

Question

lookup = numpy.array([
    [["a", "b", "c"], ["d", "e", "f"]],
    [["g", "h", "i"], ["j", "k", "l"]],
    [["m", "n", "o"], ["p", "q", "r"]],
    ])
i = numpy.array([
    [[0,0,0], [1,0,2],],
    [[2,0,1], [1,0,2],],
    ])

我想使用数组i 来访问数组lookup 中的值。 i 的尺寸为 m x n x 3（3 因为 lookup 是一个 3 维数组）。所以我想得到一个维度为 m x n 的数组，其中填充了 a[i[m][n]] 的值。所以在这种情况下预期：

out = [
["a", "i"],
["n", "i"]
]

我可以在 python 中使用 for 循环，但是我如何在 numpy 中进行呢？

Answer 1

您可以从i 中提取坐标，并使用它们来提取lookup 的子集。

import numpy 

lookup = numpy.array([
    [["a", "b", "c"], ["d", "e", "f"]],
    [["g", "h", "i"], ["j", "k", "l"]],
    [["m", "n", "o"], ["p", "q", "r"]],
    ])
i = numpy.array([
    [[0,0,0], [1,0,2],],
    [[2,0,1], [1,0,2],],
    ])
    
x = i[:,:,0]
y = i[:,:,1]
z = i[:,:,2]

out = lookup[x,y,z]

输出

[['a' 'i']
 ['n' 'i']]