python HJ35 蛇形矩阵

蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。下面是使用Python实现蛇形矩阵的代码：

def snake_matrix(n):
    matrix = [[0] * n for _ in range(n)]  # 创建一个n×n的矩阵，初始值为0
    num = 1  # 要填入矩阵的数值
    for i in range(n):
        if i % 2 == 0:
            for j in range(i + 1):
                matrix[j][i - j] = num
                num += 1
        else:
            for j in range(i + 1):
                matrix[i - j][j] = num
                num += 1
    return matrix

n = int(input("请输入一个正整数："))
matrix = snake_matrix(n)
for row in matrix:
    for val in row:
        print(val, end=" ")
    print()