如何迭代n×n矩阵并将x、y和值存储到nx3矩阵中

def iterate_matrix(n):
    matrix = [[0 for _ in range(3)] for _ in range(n*n)]  # 创建一个nx3的矩阵
    index = 0  # 矩阵索引

    for i in range(n):
        for j in range(n):
            matrix[index][0] = i  # 存储x值
            matrix[index][1] = j  # 存储y值
            matrix[index][2] = matrix_value(i, j)  # 存储值
            index += 1

    return matrix

def matrix_value(x, y):
    # 根据x和y计算矩阵中的值，这里可以根据实际需求进行具体实现
    return x + y

n = 5
result = iterate_matrix(n)
print(result)