生成蛇形矩阵——python

蛇形矩阵是由1开始的自然数依次排列成的一个上三角矩阵（如图所示）。要求输入整数n，生成并输出蛇形矩阵。

 

输入格式:

输入一个正整数n（n不大于50)。

输出格式:

输出一个n行的蛇形矩阵。每行的每两个数字之间留一个空格。

输入样例:

在这里给出一组输入。例如：

6

输出样例:

在这里给出相应的输出。例如：

1 3 6 10 15 21
2 5 9 14 20
4 8 13 19
7 12 18
11 17
16

n = int(input())
lists = [[0]*n for i in range(n)]   
lists1 = [[0]*n for i in range(n)]    
# print(lists)
lists[0][0] = 1                       
for i in range(1,n):                   
    lists[i][0] = lists[i-1][0] + i
for i in range(n):                    
    for j in range(1,n):
        lists[i][j] = lists[i][j-1]+j+i+1
for i in range(n):                   
    if i != 0:                         
        lists1[i] = [str(j) for j in lists[i][:-i]]  
    else:
        lists1[i] = [str(j) for j in lists[i]] 
    a = ' '.join(lists1[i])
    print(a)