numpy 创建单位矩阵
时间: 2023-10-10 12:04:20 浏览: 373
numpy创建单位矩阵和对角矩阵的实例
您可以使用NumPy的`eye()`函数来创建单位矩阵。`eye()`函数接受一个整数参数n,表示矩阵的维度,它会返回一个n×n的单位矩阵。
以下是使用NumPy创建3×3单位矩阵的示例代码:
```python
import numpy as np
n = 3
unit_matrix = np.eye(n)
print(unit_matrix)
```
输出结果为:
```
array([[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]])
```
希望对您有所帮助!如有其他问题,请随时提问。
阅读全文