2D陣列一維中的Numpy局部最大值

Question

我想找到二維陣列的局部最大值，但只能在一維中找到。即：

1 2 3 2 1 1 4 5 6 2 
2 2 3 3 3 2 2 2 2 2 
1 2 3 2 2 2 2 3 3 3 

將返回：

0 0 1 0 0 0 0 0 1 0 
0 0 0 1 0 0 0 0 0 0 
0 0 1 0 0 0 0 0 1 0 

這顯然是微不足道的通過數組迭代來解決，但這是緩慢的，通常是可以避免的。有沒有一個快速的方法來實現這一目標？

編輯： 我已經設計了一個更快的解決方案：

進口numpy的爲NP

testArray = np.array([[1,2,3,2,1,1,4,5,6,2],[2,2,3,3,3,2,2,2,2,2],[1,2,3,2,2,2,2,3,3,3] ]) 

leftShift = np.roll(testArray,1, axis=1) 
rightShift = np.roll(testArray,-1, axis=1) 

Max = ((testArray>leftShift) & (testArray>rightShift))*1 

print(Max) 

將返回：

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

這是除了重複讀數正確的結果。即......「13331」（最大值）與「13333789」（固定點）的區別

Answer 1

您可以通過對每一行應用有限差分梯度並檢查符號變化來解決此問題。 但是不清楚在邊界上應該做什麼。