在Numpy中制作特殊的对角矩阵

Question

我试图做一个看起来像这样的numpy数组：

[a b c       ]
[  a b c     ]
[    a b c   ]
[      a b c ]

因此，这涉及更新主对角线及其上方的两个对角线。

什么是有效的方法？

Answer 1

这是一个Toeplitz矩阵的示例-
您可以使用scipy.linalg.toeplitz以下命令构造它：

import numpy as np
from scipy.linalg import toeplitz

first_row = np.array([1, 2, 3, 0, 0, 0])
first_col = np.array([1, 0, 0, 0])

print(toeplitz(first_col, first_row))
# [[1 2 3 0 0 0]
#  [0 1 2 3 0 0]
#  [0 0 1 2 3 0]
#  [0 0 0 1 2 3]]