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

import numpy as np

def get_neighbors(arr, max_idx):
    rows, cols = arr.shape
    neighbors = []
    
    for i in range(max(-1, max_idx[0]-1), min(rows, max_idx[0]+2)):
        for j in range(max(-1, max_idx[1]-1), min(cols, max_idx[1]+2)):
            if (i != max_idx[0]) or (j != max_idx[1]):
                neighbors.append(arr[i, j])
    
    return neighbors

# 示例二维数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 找到最大值的索引
max_idx = np.unravel_index(np.argmax(arr), arr.shape)

# 获取最大值的邻居
neighbors = get_neighbors(arr, max_idx)
print("最大值的邻居:", neighbors)