python - Numpy:通过重复序列添加垫来创建上三角矩阵

Question

这个问题在这里已经有了答案:





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]

为 任何 序列

Answer 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]]