创建对角矩阵python
时间: 2023-12-29 08:05:13 浏览: 34
对角矩阵是指除了对角线上的元素外,其他元素都为0的矩阵。在Python中,可以使用numpy库创建对角矩阵。
代码示例:
```python
import numpy as np
# 创建3x3的对角矩阵
a = np.diag([1, 2, 3])
print(a)
```
输出结果:
```
[[1 0 0]
[0 2 0]
[0 0 3]]
```
在上述代码中,np.diag()函数创建了一个对角矩阵,并将其赋值给变量a。函数的参数是一个列表,列表中的元素就是对角线上的元素。
如果要创建一个更大的对角矩阵,可以使用np.eye()函数,该函数可以创建一个指定大小的单位矩阵,然后再乘以一个数得到对角矩阵。
代码示例:
```python
import numpy as np
# 创建5x5的对角矩阵
b = 2 * np.eye(5)
print(b)
```
输出结果:
```
[[2. 0. 0. 0. 0.]
[0. 2. 0. 0. 0.]
[0. 0. 2. 0. 0.]
[0. 0. 0. 2. 0.]
[0. 0. 0. 0. 2.]]
```
在上述代码中,np.eye()函数创建了一个5x5的单位矩阵,然后乘以2得到对角矩阵,并将其赋值给变量b。
相关问题
python怎么创建对角矩阵
在Python中,可以使用numpy库中的diag函数来创建对角矩阵。具体方法如下:
```python
import numpy as np
# 创建一个3x3的对角矩阵,对角线上的元素为1,2,3
diag_matrix = np.diag([1, 2, 3])
print(diag_matrix)
```
输出结果为:
```
array([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
```
其中,np.diag函数的参数为一个列表,列表中的元素为对角线上的元素。如果需要创建一个n阶的单位矩阵,可以使用np.eye(n)函数。
Python 对角矩阵
Python中可以使用NumPy库来创建对角矩阵。下面是两种创建对角矩阵的方法:
1. 使用numpy.diag函数创建对角矩阵:
```python
import numpy as np
# 创建一个3x3的对角矩阵,对角线元素为1, 2, 3
diag_matrix = np.diag([1, 2, 3])
print(diag_matrix)
```
输出:
```
[[1 0 0]
[0 2 0]
[0 0 3]]
```
2. 使用numpy.diagflat函数创建对角矩阵:
```python
import numpy as np
# 创建一个3x3的对角矩阵,对角线元素为1, 2, 3
diag_matrix = np.diagflat([1, 2, 3])
print(diag_matrix)
```
输出:
```
[[1 0 0]
[0 2 0]
[0 0 3]]
```
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)