[Numpy-discussion] can argmax be used to return row and column indices?
gregorio.bastardo at gmail.com
Fri Sep 13 06:03:47 EDT 2013
There's already been some dance around this topic, maybe you will find
the concept behind it:
2013/9/13 Mark Bakker <markbak at 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,
>> Hi Mark,
>> You're looking for "np.unravel_index" function.
>> 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).
> NumPy-Discussion mailing list
> NumPy-Discussion at scipy.org
More information about the NumPy-Discussion