[Numpy-discussion] can argmax be used to return row and column indices?
ben.root at ou.edu
Fri Sep 13 10:28:20 EDT 2013
On Fri, Sep 13, 2013 at 4:27 AM, Mark Bakker <markbak at 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 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,
What should it do when np.argmax() gets "axis=1" argument? I see confusion
occurring with parsing the returned results for arbitrary dimension inputs.
-1 on any such PR. +1 on making sure all arg*() functions have
unravel_index() very prominent in their documentation (which it does right
now for argmax()).
-------------- next part --------------
An HTML attachment was scrubbed...
More information about the NumPy-Discussion