[Numpy-discussion] is there a way to get Nd indices of max of Nd array
Chris Colbert
sccolbert@gmail....
Sun May 3 17:31:37 CDT 2009
wait, nevermind. your're right. Thanks!
On Sun, May 3, 2009 at 6:30 PM, Chris Colbert <sccolbert@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@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/d1962291/attachment.html
More information about the Numpy-discussion
mailing list