[Numpy-discussion] can argmax be used to return row and column indices?
markbak at gmail.com
Fri Sep 13 04:14:51 EDT 2013
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).
-------------- next part --------------
An HTML attachment was scrubbed...
More information about the NumPy-Discussion