[Numpy-discussion] Optimizing mean(axis=0) on a 3D array

Martin Spacek numpy at mspacek.mm.st
Sun Aug 27 08:05:21 EDT 2006


Travis Oliphant wrote:
> 
> If frameis is 1-D, then you should be able to use
> 
> temp = data.take(frameis,axis=0) 
> 
> for the first step.   This can be quite a bit faster (and is a big 
> reason why take is still around).   There are several reasons for this 
> (one of which is that index checking is done over the entire list when 
> using indexing). 
> 

Yup, that dropped the indexing step down to essentially 0 seconds.

Thanks Travis!

Martin




More information about the NumPy-Discussion mailing list