如何在单位矩阵的对角元素中插入列表元素？

Question

我有5x5单位矩阵和一个浮点数列表

 1 0 0 0 0
 0 1 0 0 0
 0 0 1 0 0
 0 0 0 1 0 
 0 0 0 0 1         L=[0.01, 0.02, 0.26, 0.03, 0.68]

我的问题是如何将列表中的元素放入单位矩阵中？在

^{pr2}$

谢谢

Answer 1

您可以使用numpy：

import numpy as np
x = np.eye(5)    # create a 5x5 identity matrix
L = [0.01, 0.02, 0.26, 0.03, 0.68]
x = x * L