[SciPy-user] How to use argmax ?

Daniel Lenski dlenski at gmail.com
Mon Oct 22 15:08:16 EDT 2007


On Mon, 22 Oct 2007 19:04:54 +0000, Daniel Lenski wrote:

> Or maybe this is what you're looking for:
> 
>>>>> ind = a.argmax(axis=0)
>>>>> a[ ind, [0,1,2] ]
> array([60, 20, 14])
> 
> ... which basically returns a[ind[0],0], a[ind[1],1], and a[ind[2],2].
> 
> Dan

Yeah, there really must be a straightforward, easy way to do this without 
explicitly listing the indices.  Hmmm...




More information about the SciPy-User mailing list