如何用python打印蛇形矩阵

要用Python打印蛇形矩阵，可以通过编写代码来实现。使用循环、索引操作、和条件判断来控制矩阵的填充顺序。这里以一个5×5的蛇形矩阵为例，详细描述如何实现。

创建一个n x n的空矩阵，并定义初始值。接着，使用循环填充矩阵，并根据当前行号决定填充方向。在偶数行从左到右填充，而在奇数行从右到左填充。

一、初始化矩阵

首先，创建一个n x n的空矩阵，并用0初始化每个元素。同时，定义起始值为1，用于填充矩阵。

def create_empty_matrix(n):
    return [[0 for _ in range(n)] for _ in range(n)]
n = 5
matrix = create_empty_matrix(n)
start_value = 1

二、填充矩阵

接下来，使用双重循环遍历矩阵。在外层循环中遍历每一行，内层循环根据当前行号决定填充方向。

for i in range(n):
    if i % 2 == 0:
        for j in range(n):
            matrix[i][j] = start_value
            start_value += 1
    else:
        for j in range(n-1, -1, -1):
            matrix[i][j] = start_value
            start_value += 1

三、打印矩阵

最后，使用循环打印矩阵。

for row in matrix:
    print(row)

完整的代码如下：

def create_empty_matrix(n):
    return [[0 for _ in range(n)] for _ in range(n)]
n = 5
matrix = create_empty_matrix(n)
start_value = 1
for i in range(n):
    if i % 2 == 0:
        for j in range(n):
            matrix[i][j] = start_value
            start_value += 1
    else:
        for j in range(n-1, -1, -1):
            matrix[i][j] = start_value
            start_value += 1
for row in matrix:
    print(row)

四、扩展与优化

上面的代码实现了基本的蛇形矩阵打印功能，接下来可以考虑一些扩展与优化。

1、动态输入矩阵大小

可以通过用户输入动态调整矩阵的大小。

n = int(input("Enter the size of the matrix: "))
matrix = create_empty_matrix(n)
start_value = 1
for i in range(n):
    if i % 2 == 0:
        for j in range(n):
            matrix[i][j] = start_value
            start_value += 1
    else:
        for j in range(n-1, -1, -1):
            matrix[i][j] = start_value
            start_value += 1
for row in matrix:
    print(row)

2、函数封装

将代码封装成函数，方便调用。

def print_snake_matrix(n):
    def create_empty_matrix(n):
        return [[0 for _ in range(n)] for _ in range(n)]
    matrix = create_empty_matrix(n)
    start_value = 1
    for i in range(n):
        if i % 2 == 0:
            for j in range(n):
                matrix[i][j] = start_value
                start_value += 1
        else:
            for j in range(n-1, -1, -1):
                matrix[i][j] = start_value
                start_value += 1
    for row in matrix:
        print(row)
n = int(input("Enter the size of the matrix: "))
print_snake_matrix(n)

五、其他蛇形填充方式

除了上述从左到右、从右到左的蛇形填充方式，还可以实现其他形式的蛇形填充，例如从上到下、从下到上的蛇形填充。下面是一个从上到下、从下到上的蛇形填充示例。

def print_vertical_snake_matrix(n):
    def create_empty_matrix(n):
        return [[0 for _ in range(n)] for _ in range(n)]
    matrix = create_empty_matrix(n)
    start_value = 1
    for j in range(n):
        if j % 2 == 0:
            for i in range(n):
                matrix[i][j] = start_value
                start_value += 1
        else:
            for i in range(n-1, -1, -1):
                matrix[i][j] = start_value
                start_value += 1
    for row in matrix:
        print(row)
n = int(input("Enter the size of the matrix: "))
print_vertical_snake_matrix(n)