Python - 将一个5x5x3 numpy数组打印为三个5x5 numpy数组？

Question

当我打印我的5x5x3 numpy阵列时，它会打印为5个5x3阵列。

我已阅读在5像素×5像素的RGB图像，并要查看的5×5阵列即是R，G和B.如何做呢？

谢谢！

Answer 1

像这样的东西应该工作：

In [1]: import numpy as np 

In [2]: a = np.array([[[1,2,3], [3,4,3]],[[5,6,3], [7,8,3]]]) 

In [3]: a.shape 
Out[4]: (2, 2, 3) 

In [4]: for i in range(a.shape[-1]): 
    ...:  print(a[:,:,i]) 
    ...: 
[[1 3] 
[5 7]] 
[[2 4] 
[6 8]] 
[[3 3] 
[3 3]]