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

这是一个Toeplitz矩阵的示例-
您可以使用

scipy.linalg.toeplitz

import numpy as npfrom scipy.linalg import toeplitzfirst_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]]