How to change numpy image?

Question

I have a numpy array image with a shape (channels, height, width), how can I change it so that it has a shape (height, width, channels)?

Answer 1

I assume it will be faster to use the numpy built-in function.

np.rollaxis(array_name,0,3).shape

Answer 2

You can use transpose()to select the order of the axes. In this case, you want:

array.transpose(1, 2, 0)

This creates a new view in the original array when possible (data will not be copied).

Answer 3

You can do:

import numpy as np
newim = np.zeros((height, width, channels))
for x in xrange(channels):
    newim[:,:,x] = im[x,:,:]