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

[Gregorio Bastardo]

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 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,
> 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 at scipy.org
> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>