Numpy：通过重复序列添加pad [重复]创建上三角矩阵

kknvjkwl 于 2023-10-19 发布在其他

关注(0)|答案(1)|浏览(90)

此问题已在此处有答案：

Generating banded matrices using numpy（3个答案）
三年前就关门了。
从给定序列创建上三角矩阵的最有效方法是什么，如下所示：
输入：

[1, 2, 3, 4, 5]

输出量：

[[1, 2, 3, 4, 5],
 [0, 1, 2, 3, 4],
 [0, 0, 1, 2, 3],
 [0, 0, 0, 1, 2],
 [0, 0, 0, 0, 1]

对于任何序列

1条答案

np.triu是您正在寻找的：它将array作为参数，并返回上三角矩阵（documentation）：

import numpy as np

seq = [1, 2, 3, 4, 5]

res = np.triu(seq)

输出量：

[[1 2 3 4 5]
 [0 2 3 4 5]
 [0 0 3 4 5]
 [0 0 0 4 5]
 [0 0 0 0 5]]