Numpy - 通過測試相鄰索引獲取索引位置

Question

我使用numpy是Python。我已預裝成numpy的2維陣列的圖像爲：

[ 
    [...], # row0 
    [...], # row1 
    [...], # row2 
    ... 
] 

我需要獲得所有像素的所有索引位置，其中任一北（只有下列之一），南，東或西相鄰像素具有一定的價值。在我的情況下，如果任何4個相鄰像素的0

Answer 1

如果是你原來的數組，定義了一堆片：

from scipy import * 

a = ones((12,22)) 
a[5,10] = a[5,12] = 0 

a_ = a[1:-1, 1:-1] 
aE = a[1:-1, 0:-2] 
aW = a[1:-1, 2:] 
aN = a[0:-2, 1:-1] 
aS = a[ 2:, 1:-1] 

a4 = dstack([aE,aW,aN,aS]) 
num_adjacent_zeros = sum(a4 == 0, axis=2) 
print num_adjacent_zeros 

ys,xs = where(num_adjacent_zeros == 1) 
# account for offset of a_ 
xs += 1 
ys += 1 

print '\n hits:' 
for col,row in zip(xs,ys): 
    print (col,row) 

的理由採取小a_的是，我不知道你想與邊緣的情況下，例如在那裏做什麼北方像素可能不存在。

我建立了一個相鄰零的數量的數組，並使用它來獲得剛好相鄰的零的位置。輸出：

[[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 1 0 1 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 1 0 2 0 1 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 1 0 1 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0] 
[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0]] 

hits: 
(10, 4) 
(12, 4) 
(9, 5) 
(13, 5) 
(10, 6) 
(12, 6) 

Answer 2

或許要做到這一點是找到所有使用類似的零的最簡單的方法：

import numpy as np 
# a is the image array 
z_indices = np.where(a == 0) 

然後就是計算相鄰指數來像素爲零（所有+ 1，-1組合爲零指數）。如果一個點與兩個不同的零像素相鄰，則必須刪除重複項。