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

Mark Bakker markbak at gmail.com
Fri Sep 13 04:14:51 EDT 2013


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/a184eaf8/attachment.html>


More information about the NumPy-Discussion mailing list