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]]