我有一个1x2x3的数组:
>>> a = np.array([[[1,2,3],[4,5,6]]])
>>> a
array([[[1, 2, 3],
[4, 5, 6]]])
>>> a.shape
(1, 2, 3)
我希望将其重塑为(3,1,2),但是让原始维度3上的元素现在沿着维度1。 我希望结果看起来像这样:
>>> new_a
array([[[1, 4]],
[[2, 5]],
[[3, 6]]])
使用reshape函数时,可以得到正确的形状,但元素顺序并不符合要求。
>>> a.reshape((3,1,2))
array([[[1, 2]],
[[3, 4]],
[[5, 6]]])
我该如何达成这个目标?