📜  Python|打印二维列表的对角线(1)

📅  最后修改于: 2023-12-03 15:19:20.205000             🧑  作者: Mango

Python | 打印二维列表的对角线

在Python中,我们可以使用嵌套列表(二维列表)来表示矩阵或表格等数据结构。当需要输出二维列表的对角线时,我们可以使用下面的方法。

# 定义一个二维列表,即矩阵
matrix = [[1, 2, 3],
          [4, 5, 6],
          [7, 8, 9]]

# 打印主对角线
for i in range(len(matrix)):
    print(matrix[i][i])

# 打印副对角线
for i in range(len(matrix)):
    print(matrix[i][len(matrix)-1-i])

在上面的代码中,我们使用range函数来生成从0到矩阵长度的整数序列,然后通过索引访问二维列表中的元素。matrix[i][i]表示主对角线上的元素,matrix[i][len(matrix)-1-i]表示副对角线上的元素。

对于任意大小的矩阵,我们都可以使用这种方法打印对角线。如果你想要将其封装成函数,可以使用下面的代码:

def print_diagonal(matrix):
    for i in range(len(matrix)):
        print(matrix[i][i])
    print()
    for i in range(len(matrix)):
        print(matrix[i][len(matrix)-1-i])

# 定义一个4x4的矩阵
matrix = [[1, 2, 3, 4],
          [5, 6, 7, 8],
          [9, 10, 11, 12],
          [13, 14, 15, 16]]

# 调用函数打印对角线
print_diagonal(matrix)

在上面的代码中,我们定义了一个名为print_diagonal的函数,它接受一个二维列表作为参数,并打印主对角线和副对角线。在函数内部,我们使用了和之前相同的方法进行打印。

最终,我们使用一个4x4的矩阵作为输入调用了print_diagonal函数,打印了主对角线和副对角线的元素。

以上就是在Python中打印二维列表的对角线的方法以及示例代码。通过这个简单的例子,我们可以看到Python中嵌套列表的用法,以及如何利用循环索引等基本语法来处理二维数组。