python - 有没有办法获得 numpy 中的最大值矩阵？

Question

问题是我有一个包含 RGB channel 的图像。我想从中获取一个矩阵，该矩阵具有跨轴图像的最大值 - 即 channel 。

例如

np.random.seed(42)
rand=np.random.randint(low=0, high=255, size=(4,8,3), dtype="uint8")
rand[0][np.newaxis,:,:]=
  [[[102, 220, 225],
    [ 95, 179,  61],
    [234, 203,  92],
    [  3,  98, 243],
    [ 14, 149, 245],
    [ 46, 106, 244],
    [ 99, 187,  71],
    [212, 153, 199]]

从中我想得到一个看起来像这样的矩阵 [[220],[179],[234],[243],[245],[244],[187],[212]]。

Argmax 为我提供了此类值的索引。

 np.argmax(exp,2) #array([[2, 1, 0, 2, 2, 2, 1, 0]], dtype=int64)

如何使用此信息来获取值本身或者是否有其他方法来获取它们？

同样的问题适用于整个矩阵，而不仅仅是一行。

Answer 1

我想我找到了答案 - 就像 np.max(rand,axis=2) 一样简单 here .

正如 @Divakar 评论中那样 - 另一种使用 rand.max(axis=-1) 的方式。