numpy python中的等效diag R函数

Question

当我使用R diag(0.03,2)时，我获得了：

     [,1] [,2]
[1,] 0.03 0.00
[2,] 0.00 0.03

如何在Python中获得相同的结果？我尝试numpy.diag(0.03,2)，但收到错误：ValueError: Input must be 1- or 2-d。与numpy.diagonal(0.03,2)

相同

感谢您的帮助

Answer 1

执行numpy.diag([.03]*2)。

numpy.diag()接收对角线组件的列表，您可以在python中将列表相乘以重复元素。