python - 如何使用在对角线上具有特定值 K 的 numpy 创建单位矩阵

我希望数组是这样的:

 array([[K., 0., 0., 0.],
           [0., K., 0., 0.],
           [0., 0., K., 0.],
           [0., 0., 0., K.]])

最佳答案

除了 liliscent answer。

如果你想在 digonal 有不同的元素，你可以创建一个对角元素列表。

x= [2,3,4]

并使用np.diag()

print(np.diag(x))

输出:

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

关于python - 如何使用在对角线上具有特定值 K 的 numpy 创建单位矩阵，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/48719867/

相关文章：

python - 如何在 Python 中将数组列表导出到 csv？