[Numpy-discussion] summing over more than one axis
greg whittier
gregwh at gmail.com
Thu Aug 19 10:57:19 EDT 2010
On Thu, Aug 19, 2010 at 10:12 AM, Angus McMorland <amcmorl at gmail.com> wrote:
> Another rank-generic approach is to use apply_over_axes (you get a
> different shape to the result this way):
>
> a = np.random.randint(20, size=(4,3,5))
> b = np.apply_over_axes(np.sum, a, [1,2]).flat
> assert( np.all( b == a.sum(axis=2).sum(axis=1) ) )
>
Thanks for the responses! This looks like what I've been looking for.
There's really a lot to numpy. It's very powerful and I feel like
I've only scratched the surface.
More information about the NumPy-Discussion
mailing list