주어진 매트릭스의 k- 이웃 N

Question

정수의 행렬의 위치 (x, y)를 제공하는 코드를 작성하려고합니다. 모든 이웃 (왼쪽, 오른쪽, 위, 아래, 아래)을 반복 할 수 있습니다. 대각선)을 다음과 같이 계산합니다.

K = 1 

01 02 03 04 05 06 
07 08 09 10 11 12 
13 14 15 16 17 18 
19 20 21 22 23 24 
25 26 27 28 29 30 

for i in range(M): 
for j in range(N): 
for ... #loop that i am writing... <br/> 
#for (0,0) is [(0,0), (0,1), (1,0), (1, 1)] 
#for (3, 3) is [(2,2), (2,3), (2,4), (3,2), (3,3), (3,4), (4,2), (4,3), (4,4) 

매트릭스의 모든 위치에 대해이 방법을 사용합니다.

팁이나 코드는 무엇입니까?

Answer 1

NumPy와 슬라이스는

>>> import numpy as np 
>>> a = np.array(range(25)).reshape((5,5)) 
>>> print a 
[[ 0 1 2 3 4] 
[ 5 6 7 8 9] 
[10 11 12 13 14] 
[15 16 17 18 19] 
[20 21 22 23 24]] 
>>> k = 1 
>>> a[2, 3] 
13 
>>> a[2-k:2+k+1, 3-k:3+k+1] 

array([[ 7, 8, 9], 
     [12, 13, 14], 
     [17, 18, 19]]) 

Be careful of edge-cases, where index -1 will wrap to the other side of the array이 아주 쉽게 만들 것입니다.