python - NumPy：如何检索多维数组中最大值的索引

Question

使用以下数组：

In [103]: da                                                                                         
Out[103]: 
array([[[ 6, 22,  3],
        [ 4,  9, 20],
        [21, 16,  0]],

       [[ 2, 25, 11],
        [ 5, 17, 18],
        [23, 13,  7]],

       [[10, 14, 26],
        [ 8,  1, 19],
        [15, 12, 24]]])

In [104]: da.shape                                                                                   
Out[104]: (3, 3, 3)

具有最大值的元素的索引可以通过以下方式确定：

In [114]: np.unravel_index(np.argmax(da), da.shape)                                                  
Out[114]: (2, 0, 2)

并检查：

In [115]: da[2, 0, 2]                                                                                
Out[115]: 26

但是，是否可以在不循环/迭代的情况下确定 9 个索引，da[:, i1, i2]其中包含 0、1 或 2的每组整数的最大值？i1i2

例如，组da[:, 0, 0]为 6、2 和 10。最大值为 10，其索引为da[2, 0, 0]。

标签： pythonnumpyindexing

Answer 1

该axis参数允许您指定单个操作轴：

i0 = np.argmax(da, axis=0)

这意味着i0是一个(3, 3)数组，其中包含每个对应i1,的最大值的索引i2。在您的示例中i1，any 的最大值是i2

da[i0[i1, i2], i1, i2]