np.diag([1,2,3,4])创建的是数组还是矩阵
时间: 2024-03-21 22:41:39 浏览: 117
numpy.diag()函数用法
np.diag([1,2,3,4])创建的是一个矩阵,具体来说是一个4行4列的对角矩阵,对角线上的元素为1,2,3,4。可以通过以下代码来验证:
```python
import numpy as np
mat = np.diag([1,2,3,4])
print(mat)
```
输出结果为:
```
array([[1, 0, 0, 0],
[0, 2, 0, 0],
[0, 0, 3, 0],
[0, 0, 0, 4]])
```
可以看到,返回的是一个矩阵,而不是一个数组。
阅读全文