Question

试图找到不等于黑色的颜色，我需要最快的方法来做到这一点。扼杀，下面的代码适用于条件！=。它不适用于>或=。我也不想对通道求和并搜索大于0的值（那太慢了）

import numpy as np
x = np.zeros(shape=[2,10,3])
x[0,0]=np.array([0,0,255])
x[0,5]=np.array([255,0,0])
x[0,8]=np.array([255,0,0])
x[0,9]=np.array([255,0,0])
x[1,2]=np.array([0,0,255])
x[1,1]=np.array([255,0,0])
x[1,9]=np.array([255,0,0])
x[1,6]=np.array([255,0,0])

indices=(np.where(np.all(x[0,:]!=[0,0,0],axis=-1),))
print('values',x[i])

>> []

感谢您提前提出建议

Answer 1

IIUC，您可以执行以下操作以查找数组中至少具有一个非零值的位置：

x[np.any(x != 0, axis = 2)]
# Or, using np.where as you were:
# np.where(np.any(x!=0, axis=2))
# (array([0, 0, 0, 0, 1, 1, 1, 1]), array([0, 5, 8, 9, 1, 2, 6, 9]))

示例：

>>> x
array([[[  0.,   0., 255.],
        [  0.,   0.,   0.],
        [  0.,   0.,   0.],
        [  0.,   0.,   0.],
        [  0.,   0.,   0.],
        [255.,   0.,   0.],
        [  0.,   0.,   0.],
        [  0.,   0.,   0.],
        [255.,   0.,   0.],
        [255.,   0.,   0.]],

       [[  0.,   0.,   0.],
        [255.,   0.,   0.],
        [  0.,   0., 255.],
        [  0.,   0.,   0.],
        [  0.,   0.,   0.],
        [  0.,   0.,   0.],
        [255.,   0.,   0.],
        [  0.,   0.,   0.],
        [  0.,   0.,   0.],
        [255.,   0.,   0.]]])

>>> x[np.any(x != 0, axis = 2)]
array([[  0.,   0., 255.],
       [255.,   0.,   0.],
       [255.,   0.,   0.],
       [255.,   0.,   0.],
       [255.,   0.,   0.],
       [  0.,   0., 255.],
       [255.,   0.,   0.],
       [255.,   0.,   0.]])

numpy，其中RGB通道大于[0,0,0]

1 个答案: