#1a) mappingNumberKey = {} nr = 1 for i in range(3): for j in range(3): mappingNumberKey[nr] = i,j nr += 1 #1b) import random as ra tall = ra.randint(1,9) #1c) import numpy as np brett = np.zeros((3,3)) r1, c1 = mappingNumberKey[tall] brett[r1][c1] = np.nan #1d) def finn_naboer(r_m, c_m, brett_m): for i in range(1,10): r, c = mappingNumberKey[i] dist = np.sqrt((r_m-r)**2 + (c_m-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)