沿某些轴的多维数组的argmax

Question

我有一个多维数组，如下所示：

my_array = np.arange(2)[:,None,None] *np.arange(4)[:, None]*np.arange(8)

我正在寻找一个多维的相当于2-D argmax的东西

特别是，我正在沿着轴= 2寻找won的argmax。我首先尝试了整形，但整形将完全破坏整个数组的原始索引信息，因此它可能不起作用。我不知道怎么做，需要你们的帮助。提前谢谢你

编辑:所需输出为：

[(0,0,0),(1,3,1),(1,3,2),(1,3,3),(1,3,4),(1,3,5),(1,3,6),(1,3,7)]

这就是沿着轴=2的最大值索引的数组

Answer 1

为了沿着3Dndarray的最后一个轴找到这样的argmax索引，我们可以沿着这些线使用一些东西-

In [66]: idx = my_array.reshape(-1,my_array.shape[-1]).argmax(0)

In [67]: r,c = np.unravel_index(idx,my_array.shape[:-1])

In [68]: l = np.arange(len(idx))

In [69]: np.c_[r,c,l]
Out[69]: 
array([[0, 0, 0],
       [1, 3, 1],
       [1, 3, 2],
       [1, 3, 3],
       [1, 3, 4],
       [1, 3, 5],
       [1, 3, 6],
       [1, 3, 7]])

为了将其扩展到通用的ndarray -

In [99]: R = np.unravel_index(idx,my_array.shape[:-1])

In [104]: np.hstack((np.c_[R],l[:,None]))
Out[104]: 
array([[0, 0, 0],
       [1, 3, 1],
       [1, 3, 2],
       [1, 3, 3],
       [1, 3, 4],
       [1, 3, 5],
       [1, 3, 6],
       [1, 3, 7]])