如何获得Numpy阵列的一列？

Question

我正在尝试获取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是三维的：第一个维度只有一个元素：包含行和列的完整数组。第二个维度是行，第三个维度是列。因此，要选择一个列，并将其嵌入最外层维度中，请使用以下命令：

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

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

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