如何将函数应用于考虑相邻单元的矩阵的每个元素

def apply_function_to_matrix(matrix):
    rows = len(matrix)
    cols = len(matrix[0])

    for i in range(rows):
        for j in range(cols):
            neighbors_sum = 0
            neighbors_count = 0

            # 遍历相邻单元
            for x in range(i-1, i+2):
                for y in range(j-1, j+2):
                    if x >= 0 and x < rows and y >= 0 and y < cols:
                        neighbors_sum += matrix[x][y]
                        neighbors_count += 1

            # 计算平均值
            average = neighbors_sum / neighbors_count

            # 更新当前元素的值
            matrix[i][j] = average

    return matrix

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

# 应用函数
result = apply_function_to_matrix(matrix)

print(result)