用python中的scipy写一个B=[[0,1,0],[1,0,1],[0,1,0]]的稀疏矩阵
时间: 2023-09-21 10:06:22 浏览: 57
14.Scipy调用curve_fit实现曲线拟合1
可以使用scipy.sparse库中的coo_matrix方法来创建稀疏矩阵。
代码如下:
```python
import numpy as np
from scipy.sparse import coo_matrix
# 定义稀疏矩阵的形状和非零元素的行列坐标及值
rows = [0, 1, 1, 2]
cols = [1, 0, 2, 1]
data = [1, 1, 1, 1]
# 创建稀疏矩阵
sparse_matrix = coo_matrix((data, (rows, cols)), shape=(3, 3))
print(sparse_matrix.toarray())
```
输出结果为:
```
[[0 1 0]
[1 0 1]
[0 1 0]]
```
其中,`rows`、`cols`和`data`分别表示非零元素的行、列坐标和值,`shape`表示稀疏矩阵的形状。最后通过`toarray()`方法将稀疏矩阵转化为密集矩阵并输出。
阅读全文