python - 在 numpy 中创建一个特殊矩阵

Question

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

你好

在我的经济学类(class)中，我们假设创建一个如下所示的数组。问题是我是经济学家而不是程序员。我们在 python 中使用 numpy。我们的教授说大学并没有让我们为现实世界做好准备，而是希望我们学习编程(这是一件好事)。我们不允许使用任何包，必须提供原始代码。有没有人知道如何制作这个矩阵。我花了几个小时尝试代码和浏览互联网寻求帮助，但都没有成功。

Answer 1

这种矩阵称为Toeplitz matrix或常数对角矩阵。知道这一点会让你找到 scipy.linalg.toeplitz :

import scipy.linalg
scipy.linalg.toeplitz([1, 0, 0, 0], [1, 2, 3, 0, 0, 0])

=>

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