重建numpy数组,需要更快的方法

时间:2016-11-03 22:27:55

标签: python arrays numpy

我有图像numpy数组(640,480,3),其中X,Y是坐标,(255 255 255)是点的颜色掩码

我尝试获得新的2d数组(x,y),其中X和Y是当颜色>时点的坐标。零..

我试试这段代码 并且它可以工作,但它需要太多的处理器时间

<ng-content>

1 个答案:

答案 0 :(得分:2)

您可以对最后一个轴的第一个元素进行切片,将其与0进行比较,然后使用np.argwhere获取这些索引,即xy坐标在(N,2)形状的阵列中。

因此,实施将简单地 -

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