[Numpy-discussion] is there a way to get Nd indices of max of Nd array

[Stéfan van der Walt]

2009/5/3  <josef.pktd@gmail.com>:
>>>> factors = np.random.randint(5,size=(4,3))
>>>> factors
> array([[1, 1, 3],
>       [0, 2, 1],
>       [4, 4, 1],
>       [2, 2, 4]])
>>>> factors.max()
> 4
>>>> np.argmax(factors)
> 6
>>>> np.nonzero(factors==factors.max())
> (array([2, 2, 3]), array([0, 1, 2]))

Since you have more than one maximum here, you have to do something
like Josef outlined above.  If you only want the indices of the first
maximum argument, you can use

np.unravel_index(6, factors.shape)

Regards
Stéfan