Python 生成蛇形数组/对角线数组

问题描述

1. 生成一个m * n的二维数组，元素值为1，2，3…按对角线顺序由下向上排列，如：

1，3，6，9
2，5，8，11
4，7，10，12

2. 生成一个m * n的二维数组，元素值为1，2，3…按蛇形顺序排列，如：

1，2，6，7
3，5，8，11
4，9，10，12

思路

数组同一条斜线上的元素，其下标和相等，从0到m+n-2递增。

遍历0到m+n-2下标和，每次填入一条对角线上的值。

对角线数组方向一致，很容易生成；

蛇形数组交替改变方向，可将每两次循环视为一组。

代码

对角线数组：

import numpy as np
m = 3 #行数
n = 4 #列数
list = np.zeros((m, n), dtype=np.int)