[Numpy-discussion] Vectorizing computation

Francesc Alted faltet at gmail.com
Fri Feb 13 05:51:48 EST 2015


Hi,

I would like to vectorize the next computation:

nx, ny, nz = 720, 180, 3
outheight = np.arange(nz) * 3
oro = np.arange(nx * ny).reshape((nx, ny))

def compute1(outheight, oro):
    result = np.zeros((nx, ny, nz))
    for ix in range(nx):
        for iz in range(nz):
            result[ix, :, iz] = outheight[iz] + oro[ix, :]
    return result

I think this should be possible by using an advanced use of broadcasting in
numpy.  Anyone willing to post a solution?

Thanks,
-- 
Francesc Alted
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20150213/d91f4770/attachment.html>


More information about the NumPy-Discussion mailing list