获取二维numpy数组中最大值的邻居

我有一个形状为10x10的2D数组，我需要在2D数组中找到最大值的邻居。我的代码是：

import numpy as np
array = np.random.randint(-10,10, size=(10,10))

max_index = np.argmax(array)

def get_coordinate_i(value):
        i = 0
        if value % 10 >= 1:
            i = value // 10
        return i

j = max_index - 10\*et_coordinate_i(max_index)
i = get_coordinate_i(max_index)

def fun(i, j, array):
    arr = np.array([])
    if i!=0 and j!=0 and i!=9 and j!=9:
        for val in range(-1,2):
            for val2 in range(-1,2):
                arr = np.append(arr, array[i+val][i+val2])
    return np.reshape(arr, (3,3))

不过，它并非每次都能正常运行。

问题来源：stackoverflow