How to I efficiently transform a numpy image of shape (w, h, 3) to (w,h,5) which has r,g,b,x,y in third axis?

Question

Without using for loop.

How can I attach the x,y coordinate on to each pixel of a rgb image in numpy?

such that

image[0,0,:] = (r,g,b,x,y) where x,y is the coordinate of the pixel

See [here](http://stackoverflow.com/q/42302348/7207392) for how to create the x, y, bits. — Paul Panzer, Mar 11 '17 at 08:10

score 2 · Accepted Answer · answered Mar 11 '17 at 05:59

2

Suppose rgb and xy are your (w,h,3) and (w,h,2) arrays, respectively. Then you can concatenate them along the third axis:

image = np.concatenate((rgb, xy), axis=2)

answered Mar 11 '17 at 05:59

DYZ

1

And you can use `numpy.indices` with `numpy.moveaxis` to get the `xy` you want – Liteye Mar 11 '17 at 06:03

1 Answers1