如何在python中将图像从（n，m，4）重塑为（n，m，3）？

Question

我有一个像下面这样的图像值：

A = [[[  7,  23,   0, 255],

         [ 14,  25,   1, 255],

         [ 23,  26,   1, 255],

         [ 49,  41,  33, 255],

         [ 43,  37,  30, 255],

         [ 42,  33,  25, 255]],

        [[ 59,  69,  59, 255],

         [ 77,  83,  72, 255],

         [ 86,  91,  84, 255],

         [ 16,  16,   8, 255],

         [  9,   7,   2, 255],

         [  1,   0,   0, 255]]]

print(np.array(A).shape) ## (3,6,4)

我想重塑它，就像(3,6,3)我想跳过255矩阵一样。

我能怎么做？