There's already been some dance around this topic, maybe you will find the concept behind it: http://search.gmane.org/?query=unravel_index&group=gmane.comp.python.numeric.general 2013/9/13 Mark Bakker <markbak@gmail.com>:

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

_______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@scipy.org http://mail.scipy.org/mailman/listinfo/numpy-discussion