我假设你的数组看起来像:
|(HUE)(VALUE)(CLASS)
row/col| 0 1 2
-------+-----------------
0 | 0 1 2
1 | 3 4 5
2 | 6 7 8
. | . . .
. | . . .
3599999| . . .
这是示例代码。为了简单起见,我将大小 3600000 更改为 5。
a = np.array(xrange(5 * 3))
a.shape = (5, 3)
现在阵列a
看起来像这样:
array([[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8],
[ 9, 10, 11],
[12, 13, 14]])
如果你想与HUE=9
,这样做:
a[np.where(a[:,0] == 9)]
#array([[ 9, 10, 11]])
如果你想与VALUE=4
,这样做:
a[np.where(a[:,1] == 4)]
#array([[3, 4, 5]])
如果你想与HUE=0
and VALUE=1
,这样做:
a[np.where((a[:,0] == 0) * (a[:,1] == 1))]
#array([[0, 1, 2]])