某二维数组存放的数据构成一个n*n的方阵，其中n<=5。写程序，从键盘输入n的值（n<=5），该n*n矩阵中各元素的值按下面的公式计算： a[i][j] = i * n + j + 1 其中，a[i][j]表示第i行第j列的元素。要求分别输出该矩阵和它的转置矩阵。

n = int(input("请输入n的值（n<=5）："))
matrix = [[i*n + j + 1 for j in range(n)] for i in range(n)]

print("原矩阵：")
for i in range(n):
    for j in range(n):
        print(matrix[i][j], end="\t")
    print()

transpose_matrix = [[matrix[j][i] for j in range(n)] for i in range(n)]
print("转置矩阵：")
for i in range(n):
    for j in range(n):
        print(transpose_matrix[i][j], end="\t")
    print()

输出：

请输入n的值（n<=5）：3
原矩阵：
1	2	3	
4	5	6	
7	8	9	
转置矩阵：
1	4	7	
2	5	8	
3	6	9