[Numpy-discussion] Broadcasting question

[Charles R Harris]

On Thu, May 1, 2008 at 10:58 PM, Chris.Barker <Chris.Barker@noaa.gov> wrote:

> Hi all,
>
> I have a n X m X 3 array, and I a n X M array. I want to assign the
> values in the n X m to all three of the slices in the bigger array:
>
> A1 = np.zeros((5,4,3))
> A2 = np.ones((5,4))
> A1[:,:,0] = A2
> A1[:,:,1] = A2
> A1[:,:,2] = A2
>

A1 += A2[:,:,newaxis] is one way. Or you could just start by stacking copies
of A2 and swapping axes around:

In [5]: A1 = array([A2.T]*3).T

In [6]: A1.shape
Out[6]: (4, 5, 3)

Chuck
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://projects.scipy.org/pipermail/numpy-discussion/attachments/20080501/ac977182/attachment.html