在三维 ndarray 的最后一个维度上棘手的 numpy argmax

Question

if 有一个形状数组 (9,1,3(。

array([[[  6,  12, 108]],
   [[122, 112,  38]],
   [[ 57, 101,  62]],
   [[119,  76, 177]],
   [[ 46,  62,   2]],
   [[127,  61, 155]],
   [[  5,   6, 151]],
   [[  5,   8, 185]],
   [[109, 167,  33]]])

我想找到第三维的 argmax 索引，在这种情况下它将是 185，所以索引 7。

我想解决方案与重塑有关，但我无法理解它。感谢您的任何帮助！

Answer 1

我不确定它有什么棘手之处。但是，沿最后一个轴获取最大元素索引的一种方法是使用 np.max 和np.argmax，如下所示：

# find `max` element along last axis 
# and get the index using `argmax` where `arr` is your array
In [53]: np.argmax(np.max(arr, axis=2))
Out[53]: 7

或者，正如@PaulPanzer在他的评论中所建议的那样，您可以使用：

In [63]: np.unravel_index(np.argmax(arr), arr.shape)
Out[63]: (7, 0, 2)
In [64]: arr[(7, 0, 2)]
Out[64]: 185

Answer 2

你可能必须这样做：

data = np.array([[[  6,  12, 108]],
   [[122, 112,  38]],
   [[ 57, 101,  62]],
   [[119,  76, 177]],
   [[ 46,  62,   2]],
   [[127,  61, 155]],
   [[  5,   6, 151]],
   [[  5,   8, 185]],
   [[109, 167,  33]]])
np.argmax(data[:,0][:,2])
 7