Re: [Numpy-discussion] Dimension reduction

[Charles R H. <cha...@gm...>]

On 10/19/06, David Huard <dav...@gm...> wrote:
>
> Hi,
>
> Is there an elegant way to reduce an array but conserve the reduced
> dimension ?
>
> Currently,
> >>> a = random.random((10,10,10))
> >>> a.sum(1).shape
> (10,10)
>
> but i'd like to keep (10,1,10) so I can do a/a.sum(1) directly.


In [8]: a.sum(1)[:,newaxis,:].shape
Out[8]: (10, 1, 10)

Don't know if this is as universal as you want, but it works for this.
Robert's answer is more general.

Chuck