numpy生成对角阵
时间: 2023-12-14 17:33:43 浏览: 80
numpy创建单位矩阵和对角矩阵的实例
以下是使用numpy生成对角阵的两种方法:
1.使用numpy.diag函数生成对角阵:
```python
import numpy as np
# 生成对角阵
arr = np.diag([1, 2, 3])
print(arr)
```
输出:
```
array([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
```
2.使用numpy.diagflat函数生成对角阵:
```python
import numpy as np
# 生成对角阵
arr = np.diagflat([[1, 2], [3, 4]])
print(arr)
```
输出:
```
array([[1, 0, 0, 0],
[0, 2, 0, 0],
[0, 0, 3, 0],
[0, 0, 0, 4]])
```
阅读全文