[Numpy-discussion] is there a way to get Nd indices of max of Nd array

Chris Colbert sccolbert@gmail....
Sun May 3 17:30:37 CDT 2009


but this gives me just the locations of the column/row maximums.

I need the (x,y) location of the array maximum.

Chris

On Sun, May 3, 2009 at 4:34 PM, <josef.pktd@gmail.com> wrote:

>  On Sun, May 3, 2009 at 3:30 PM, Chris Colbert <sccolbert@gmail.com>
> wrote:
> > my case is only for 2d, but should apply to Nd as well.
> >
> > It would be convienent if np.max would return a tuple of the max value
> and
> > its Nd location indices.
> >
> > Is there an easier way than just using the 1d flattened array max index
> > (np.argmax) and calculating its corresponding Nd location?
> >
>
> >>> factors = np.random.randint(5,size=(4,3))
> >>> factors
> array([[1, 1, 3],
>       [0, 2, 1],
>       [4, 4, 1],
>       [2, 2, 4]])
> >>> factors.max()
> 4
> >>> np.argmax(factors)
> 6
> >>> np.nonzero(factors==factors.max())
> (array([2, 2, 3]), array([0, 1, 2]))
>
> Josef
> _______________________________________________
> Numpy-discussion mailing list
> Numpy-discussion@scipy.org
> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://mail.scipy.org/pipermail/numpy-discussion/attachments/20090503/9073824e/attachment.html 


More information about the Numpy-discussion mailing list