numpy生成对角矩阵
时间: 2023-10-09 18:05:39 浏览: 186
numpy创建单位矩阵和对角矩阵的实例
可以使用numpy库的diag函数生成对角矩阵。例如,生成一个3x3的对角矩阵,对角线元素为1、2、3,可以使用如下代码:
```python
import numpy as np
diag_array = np.array([1, 2, 3])
diag_matrix = np.diag(diag_array)
print(diag_matrix)
```
输出结果为:
```
[[1 0 0]
[0 2 0]
[0 0 3]]
```
其中,np.array([1, 2, 3])创建了一个一维数组,表示对角线元素的值,np.diag(diag_array)则将这个一维数组转换为对应的对角矩阵。
阅读全文