#1a)
mappingNumberKey = {}

ant_rekker = 5
ant_kol = 4

nr = 1

for i in range(ant_rekker):
    for j in range(ant_kol):
         mappingNumberKey[nr] = i,j
         nr += 1
            

#1b)
import random as ra

tall = ra.randint(1,ant_rekker*ant_kol)
    

#1c)
import numpy as np

brett = np.zeros((ant_rekker,ant_kol))

r1, c1 = mappingNumberKey[tall]

brett[r1][c1] = np.nan


 
#1d) 
     
def finn_naboer(r_1, c_1, brett_m):
       
   for i in range(1,ant_rekker*ant_kol + 1):
      r, c = mappingNumberKey[i]
      dist = np.sqrt((r_1-r)**2 + (c_1-c)**2)
      if dist <= np.sqrt(2) and dist > 0:
         brett_m[r][c] = 1        
   
   return brett_m
     
brett = finn_naboer(r1, c1, brett) 

print(brett)