Python中的2D局部最大值和最小值

import numpy as np

def find_local_extrema(matrix):
    # 将二维数组转换为NumPy数组
    arr = np.array(matrix)
    
    # 获取数组的行数和列数
    rows, cols = arr.shape
    
    # 用于存储局部最大值和最小值的列表
    local_maxima = []
    local_minima = []
    
    # 遍历数组中的每个元素
    for i in range(1, rows-1):
        for j in range(1, cols-1):
            # 获取当前元素的值
            current_value = arr[i, j]
            
            # 获取当前元素周围的8个相邻元素的值
            neighbors = [arr[i-1, j-1], arr[i-1, j], arr[i-1, j+1],
                         arr[i, j-1], arr[i, j+1],
                         arr[i+1, j-1], arr[i+1, j], arr[i+1, j+1]]
            
            # 检查当前元素是否为局部最大值或最小值
            if current_value > max(neighbors):
                local_maxima.append((i, j))
            elif current_value < min(neighbors):
                local_minima.append((i, j))
    
    return local_maxima, local_minima