Question

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

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

我正在寻找二维argmax的多维等效物

尤其是，我在沿着轴= 2寻找最大值的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

要在3D ndarray的最后一个轴上找到这样的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]])

argmax用于沿某个轴的多维数组

1 个答案: