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)
list[0][1] = 1
count = 1
#k记录下标和。每个斜线上下标和相等，从0到m+n-2递增
for k in range(m+n-1):
    for i in range(max(k-m+1, 0), min(k+1, n)):
        list[k-i][i] = count
        count += 1
print(list)

蛇形数组：

import numpy as np
m = 3 #行数
n = 4 #列数
list = np.zeros((m, n), dtype=np.int)
count = 1
k = 0 #记录下标和。每个斜线上下标和相等，从0到m+n-2递增
while k<m+n-1:
    for i in range(max(k-m+1, 0), min(k+1, n)):
        list[k-i][i] = count
        count += 1
    k += 1
    if k<m+n-1:
        for i in range(max(k-n+1, 0), min(k+1, m)):
            list[i][k-i] = count
            count += 1
    k += 1
print(list)