如何将一维 numpy 数组转换为下三角矩阵？

Question

我有一个 numpy 数组，例如：

np.array([1,2,3,4])

我想把它转换成一个下三角矩阵，比如

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

，没有for循环....我该怎么做？

Answer 1

Michael Szczesny在评论中提出的类似解决方案可以是：

b = np.arange(a.size)
result = np.tril(np.take(a, b - b[:,None] + a.size - 1, mode='clip'))

结果是：

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