[Numpy-discussion] np.where and broadcasting

[Pauli Virtanen]

10.10.2011 23:02, Jesper Larsen kirjoitti:
> Hi numpy-users
>
> I have a 2d array of shape (ny, nx). I want to broadcast (copy) this
> array to a target array of shape (nt, nz, ny, nz) or (nt, ny, nx) so
> that the 2d array is repeated for each t and z index (corresponding to
> nt and nz). I am not sure how to do this (generic solution, for
> different target array shapes). Can anyone help?

assert a.shape == (nt, nz, ny, nx) or a.shape == (nt, ny, nx)
assert b.shape == (ny, nx)
a[...] = b