重建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)