reshape array in numpy,"data = np.transpose(data, (0, 3, 1, 2))"