[Numpy-discussion] can argmax be used to return row and column indices?

[Mark Bakker]

Thanks, Gregorio.
I would like it if argmax had a keyword option to return the row,column
index automatically (or whatever the dimension of the array).
Afterall, argmax already knows the shape of the array.
Calling np.unravel_index( np.argmax( A ) ) seems unnecessarily long. But it
works well though!
I am not sure that such a PR would get much support....
Thanks again,
Mark



Hi Mark,
>
> You're looking for "np.unravel_index" function.
>
> Cheers,
> Gregorio



>

Hello list,
>
> I am trying to find the indices of the maximum value in a 2D array.
> argmax works fine, but returns the index in the flattened array.
> That is often not very convenient.
> Is there a function that returns the index of the row and column?
> Or can the index of the flattened array easily be converted to the row can
> column indices (I know, I can write a function for that, but I figure numpy
> already has one).
>
> Thanks,
>
> Mark
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20130913/30926b41/attachment.html>