in range(-1,1):
array[index+i][index+j]
он с таким подходом крутить циклы в питоне и за день не обойдет) в np есть свои реализации для 2d, например https://glowingpython.blogspot.com/2012/04/k-nearest-neighbor-search.html
Обсуждают сегодня