Question

我正在尝试对三个2x2矩阵的阵列进行整形，即形状为（3,2,2）。

a = np.array([[[a1,a2],[a3,a4]],
             [[b1,b2],[b3,b4]],
             [[c1,c2],[c3,c4]]])

此形状数组（2,2,3）：

[[[a1,b1,c1],[a2,b2,c2]],
 [[a3,b3,c3],[a4,b4,c4]]])

常规np.reshape(a, (2,2,3))返回此数组：

[[[a1, a2, a3],[a4, b1, b2]],
 [[b3, b4, c1],[c2, c3, c4]]]

和np.reshape(a, (2,2,3), order = 'F')带来了这一点：

[[[a1, b3, c2],[c1, a2, b4]],
 [[b1, c3, a4],[a3, b2, c4]]]

如何重塑初始数组以获得所需的东西？

这是为了与matplotlib.pyplot.imshow一起使用，其中三个初始矩阵对应于三种颜色“ RGB”，并且每个元素都是[0,1]范围内的浮点数。因此，如果有更好的方法，我将很高兴知道。

Answer 1

我们只需要置换轴。这样做的两种方法。

使用np.transpose-

a.transpose(1,2,0) # a is input array
# or np.transpose(a,(1,2,0))

我们也可以使用np.moveaxis-

np.moveaxis(a,0,2)    # np.moveaxis(a, 0, -1)

样品运行-

In [157]: np.random.seed(0)

In [158]: a = np.random.randint(11,99,(3,2,2))

In [159]: a
Out[159]: 
array([[[55, 58],
        [75, 78]],

       [[78, 20],
        [94, 32]],

       [[47, 98],
        [81, 23]]])

In [160]: a.transpose(1,2,0)
Out[160]: 
array([[[55, 78, 47],
        [58, 20, 98]],

       [[75, 94, 81],
        [78, 32, 23]]])

Answer 2

您想要的是转置：

onCharacteristicChanged

输出：

a = np.array([[['a1', 'a2'],['a3', 'a4']],
             [['b1', 'b2'],['b3', 'b4']],
             [['c1', 'c2'],['c3', 'c4']]])

print(np.rollaxis(a, 1, 3).T)

如何以我需要的方式重塑此数组？

2 个答案: