如何生成特定形式的NxN矩阵

def generate_matrix(n):
    matrix = [[0] * n for _ in range(n)]  # 创建一个NxN的全零矩阵

    for i in range(n):
        for j in range(n):
            if i == j:
                matrix[i][j] = 1  # 对角线上的元素为1
            elif i + j == n - 1:
                matrix[i][j] = 2  # 反对角线上的元素为2

    return matrix

n = 5  # 设置矩阵的维度
matrix = generate_matrix(n)
print(matrix)