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

Chris Colbert sccolbert at gmail.com
Sun May 3 18:31:37 EDT 2009


wait, nevermind. your're right. Thanks!

On Sun, May 3, 2009 at 6:30 PM, Chris Colbert <sccolbert at gmail.com> wrote:

> 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 at gmail.com> wrote:
>
>>  On Sun, May 3, 2009 at 3:30 PM, Chris Colbert <sccolbert at 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 at scipy.org
>> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>>
>
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20090503/d1962291/attachment.html>


More information about the NumPy-Discussion mailing list