python实现二维数组的对角线遍历

对于在Python中实现对角线遍历的问题，我们可以采用以下方法：

创建一个二维数组

可以使用列表嵌套列表或NumPy库中的ndarray来创建一个二维数组。举个例子，如果我们要创建一个大小为3 x 3的矩阵，那么使用列表嵌套列表的方法可以这样写：

matrix = [
  [1,2,3],
  [4,5,6],
  [7,8,9]
]

如果我们要使用NumPy来创建一个大小为3 x 3的矩阵，可以这样写：

import numpy as np

matrix = np.array([[1,2,3], [4,5,6], [7,8,9]])

对角线遍历

接下来，我们需要对二维数组进行对角线遍历。具体来说，我们需要分别遍历主对角线和副对角线。我们可以定义两个不同的变量来表示行和列的指针。对于主对角线，初始时，行和列的指针都是0，每次遍历一次，行和列的指针都加1；对于副对角线，初始时，行的指针是0，列的指针是矩阵的列数减1，每次遍历一次，行的指针加1，列的指针减1。遍历过程中，我们需要注意边界情况，以防止数组越界。

具体代码实现如下：

def diagonal_traverse(matrix):
    m = len(matrix)
    n = len(matrix[0])
    res = [0] * (m * n)
    row, col = 0, 0
    for i in range(m * n):
        res[i] = matrix[row][col]
        if (row + col) % 2 == 0:  # move up
            if col == n - 1:
                row += 1
            elif row == 0:
                col += 1
            else:
                row -= 1
                col += 1
        else:  # move down
            if row == m - 1:
                col += 1
            elif col == 0:
                row += 1
            else:
                row += 1
                col -= 1
    return res

我们可以使用以上函数来对之前提到的两个例子进行遍历：

例子1

matrix = [
  [1,2,3],
  [4,5,6],
  [7,8,9]
]
print(diagonal_traverse(matrix)) #[1, 2, 4, 7, 5, 3, 6, 8, 9]

例子2

matrix = [
  [1,2],
  [3,4]
]
print(diagonal_traverse(matrix)) #[1, 2, 3, 4]

以上就是Python实现二维数组的对角线遍历的完整攻略了。