python中的数组整形

Question

如何将(4,n,m)大小的numpy数组重塑为(n*m,4)数组？在

动机来自于对图像的处理，我想创建一个矩阵，其中的行包含每个像素的RGBA值。我试过np.array.reshape(n*m,4)，但它做不到。可能与movaxis或{}结合使用？在

Answer 1

最简单的方法是首先将其重塑为(4, n*m)，然后交换第一个和最后一个轴（例如使用^{}或简单地使用.T属性）：

arr = arr.reshape(4, n*m).T   # assuming your array is called "arr"

例如：

^{pr2}$