不使用循环实现：输入为正整数n，返回的数组构造方式如下：第1行填入1个1，第2行在上一行填入位置的下一列连续填入2个2，第3行在第二行最后一个填入位置的下一列连续填入3个3，...

本文链接：https://blog.csdn.net/weixin_45735391/article/details/126803987

设计一个生成二维NumPy数组的函数get_res()，其输入为正整数n，返回的数组构造方式如下：第1行填入1个1，第2行在上一行填入位置的下一列连续填入2个2，第3行在第二行最后一个填入位置的下一列连续填入3个3，…，第n行在第n-1行最后一个填入位置的下一列连续填入n个n。
不使用循环实现

思路

第一步
$\begin{pmatrix} 1 \\ 2 \\ 3 \end{pmatrix} * \begin{pmatrix} 1 & 1 & 1 & 1 & 1 & 1 \\ 1 & 1 & 1 & 1 & 1 & 1 \\ 1 & 1 & 1 & 1 & 1 & 1 \end{pmatrix} = \begin{pmatrix} 1 & 1 & 1 & 1 & 1 & 1 \\ 2 & 2 & 2 & 2 & 2 & 2 \\ 3 & 3 & 3 & 3 & 3 & 3 \end{pmatrix}$
第二步
$\begin{pmatrix} 1 \\ 2 \\ 3 \end{pmatrix} * \begin{pmatrix} 1 & 1 & 1 \\ 1 & 1 & 1 \\ 1 & 1 & 1 \end{pmatrix} = \begin{pmatrix} 1 & 1 & 1 \\ 2 & 2 & 2 \\ 3 & 3 & 3 \end{pmatrix} ， L2 = L1的转置 = \begin{pmatrix} 1 & 2 & 3 \\ 1 & 2 & 3 \\ 1 & 2 & 3 \end{pmatrix}$

$\begin{pmatrix} 1 & 0 & 0 \\ 2 & 2 & 0 \\ 3 & 3 & 3 \end{pmatrix}$

$\begin{pmatrix} 1 & 0 & 0 & 2 & 2 & 0 & 3 & 3 & 3 \end{pmatrix}$

$\begin{pmatrix} 3 & 3 & 3 & 2 & 2 & 1 & 0 & 0 & 0 \end{pmatrix}$

$\begin{pmatrix} 1 & 2 & 2 & 3 & 3 & 3 \end{pmatrix}$

$\begin{pmatrix} 1 & 2 & 2 & 3 & 3 & 3 \\ 1 & 2 & 2 & 3 & 3 & 3 \\ 1 & 2 & 2 & 3 & 3 & 3 \end{pmatrix}$

第三步
$\begin{pmatrix} 1 & 0 & 0 & 0 & 0 & 0 \\ 0 & 2 & 2 & 0 & 0 & 0 \\ 0 & 0 & 0 & 3 & 3 & 3 \end{pmatrix}$

import numpy as np
def get_res(n):
    m = np.array(range(n+1)).sum()#列数
    r1 = np.ones((n,m))#生成m*n的全1矩阵
    r = np.array(range(1,n+1)).reshape((n,-1))#生成1到n的列向量
    A = r * r1
    #print(A)

    #生成1 2 2 3 3 3 4 4 4 4.....的序列
    L = np.array(range(1,n+1))
    L1 = L*np.ones((n,n))
    L2 = L1.T#生成第i行全为i+1的矩阵

    L3 = np.where(L1 > L2, 0, L2)#把右上部分变为0
    L3 = L3.reshape(-1)

    L4 = L3.tolist()
    L4.sort(reverse=True)

    m = np.array(range(n+1)).sum()
    L5 = L4[:m]#去掉0元素
    L5.sort()

    B = np.array(L5*n).reshape((n,-1))
    #print(B)

    res = np.where(A == B, A, 0)
    return res

n = 5
get_res(n)

输出：

array([[1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
       [0., 2., 2., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
       [0., 0., 0., 3., 3., 3., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0., 0., 4., 4., 4., 4., 0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 5., 5., 5., 5., 5.]])