重建numpy的阵列，需要更快的方法

Question

我有图像numpy的阵列（640,480,3），其中X，Y坐标是和（255,255,255）是彩色面具点

我试图让新的二维阵列（ X，Y），其中X和Y是坐标点时颜色>零..

我试试这个代码 和它的作品，但它需要太多的处理器时间

enter code here 
for x in range(edges.shape[0]): 
    for y in range(edges.shape[1]): 
      if edges[[x],[y],[0]]!=0: 
      new.append([x,y]) 

Answer 1

你可以切片最后一个轴的第一个元素，将其与0进行比较，然后使用np.argwhere以获得这些指数，这将是,y坐标在(N,2)形状的阵列。

因此，实现起来简单 -

new = np.argwhere(edges[...,0]!=0)