计算矩阵中某些元素的所有直接周围元素的和

def get_surrounding_sum(matrix, row, col):
    rows = len(matrix)
    cols = len(matrix[0])
    surrounding_sum = 0
    
    for i in range(max(0, row-1), min(rows, row+2)):
        for j in range(max(0, col-1), min(cols, col+2)):
            if i != row or j != col:
                surrounding_sum += matrix[i][j]
    
    return surrounding_sum

# 示例矩阵
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 计算元素 (1, 1) 的直接周围元素的和
result = get_surrounding_sum(matrix, 1, 1)
print(result)  # 输出：40