pythin中numpy的array数组求其对应的对角矩阵的问题

• Z时代
• 2024-01-10
• 分类：技术分享

1.问题描述：例如,通过矩阵运算后得到某列向量a，我想要的结果是以a为对角线的对角矩阵，但np.diag(a)得不到我想要的结果。此时应该怎么做呢？

2.代码

import numpy as np
a = np.array([[1],[2],[3]])
np.diag(a)
b = a.T
np.diag(b)

回答：

a = np.array([1,2,3])
np.diag(a)

回答：

np.diag(a.flat)

以上是 pythin中numpy的array数组求其对应的对角矩阵的问题 的全部内容， 来源链接： utcz.com/a/158658.html