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

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

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

Hi Mark, You're looking for np.unravel_index function. Cheers, Gregorio 2013/9/13 Mark Bakker mark...@gmail.com: 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

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

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

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

There's already been some dance around this topic, maybe you will find the concept behind it: http://search.gmane.org/?query=unravel_indexgroup=gmane.comp.python.numeric.general 2013/9/13 Mark Bakker mark...@gmail.com: Thanks, Gregorio. I would like it if argmax had a keyword option to

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

On Fri, Sep 13, 2013 at 4:27 AM, Mark Bakker mark...@gmail.com wrote: 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