如何獲得Numpy數組的一列？

Question

我triyng得到NumPy的（RGB）陣列的3列：

print px 
[[[ 0 0 0] 
    [255 255 255] 
    [255 0 0] 
    [ 0 255 0] 
    [ 0 0 255]]] 

print px[:,0] 
print px[:,1] 
print px[:,2] 

[[0 0 0]] 
[[255 255 255]] 
[[255 0 0]] 

，但我希望你能幫助我獲得了R，G和B樣

[[0 255 255 0 0]] 
[[0 255 0 255 0]] 
[[0 255 0 0 255]] 

？

謝謝

雨果

Answer 1

你的陣列px是三維的：所述第一維具有隻是一個單一的元件：含有行和colums完整陣列。第二個維度是行，第三個是colums。因此，要選擇一列，並將其嵌入最外層維度，請使用以下代碼：

>>> print px[:,:,0] 
[[ 0 255 255 0 0]] 

>>> print px[:,:,1] 
[[ 0 255 0 255 0]] 

>>> print px[:,:,2] 
[[ 0 255 0 0 255]]