[Numpy-discussion] unclear output format for numpy.argmax()
toddrjen at gmail.com
Tue Mar 12 03:48:36 EDT 2013
On Tue, Mar 12, 2013 at 8:20 AM, soumen ganguly
<soumendotganguly at gmail.com>wrote:
> There are some doubts that i have regarding the argmax() method of
> numpy.As described in reference doc's of numpy,argmax(axis=None,out=None)
> returns the indices of the maximum value along the given axis(In this case
> 0 is default).
> So, i tried to implement the method to a 2d array with elements
> say,[[1,2,3],[4,5,6]] along the axis 1.The output to this code is [2,2] and
> when i implement it along the axis 0,it outputs [1,1,1].I dont see the
> connection to this output with the scope of argmax method.
> I would appreciate a detailed insight to the argmax method.
I am not sure I understand the question. For axis 0 (the "outer" dimension
in the way it is printed) the things being compared are argmax([1, 4]),
argmax(([2, 5]), and argmax([3, 6]).. Amongst those, the second (index 1)
is higher in each case, so it returns [1, 1, 1]. With axis 1 (the "inner"
dimension in the way it is printed) , the things being compared are
argmax([1, 2, 3]) and argmax([4, 5, 6]). In both case the third (index 2)
is the highest, so it returns [2, 2]. What is unexpected about this?
-------------- next part --------------
An HTML attachment was scrubbed...
More information about the NumPy-Discussion