python稀疏布阵
时间: 2023-11-08 16:02:09 浏览: 75
稀疏矩阵是指大部分元素为零的矩阵。在Python中,可以使用scipy库的sparse模块来处理稀疏矩阵。下面是一个简单的例子,展示如何创建和操作稀疏矩阵:
```python
import numpy as np
from scipy.sparse import csr_matrix
# 创建稀疏矩阵
data = np.array([1, 2, 3, 4, 5, 6]) # 非零元素
row = np.array([0, 0, 1, 1, 2, 2]) # 行索引
col = np.array([0, 2, 1, 2, 0, 2]) # 列索引
sparse_matrix = csr_matrix((data, (row, col)), shape=(3, 3))
print(sparse_matrix.toarray())
```
输出结果:
```
[[1 0 2]
[0 3 4]
[5 0 6]]
```
这里通过`csr_matrix`函数创建了一个3x3的稀疏矩阵,并将非零元素、行索引和列索引作为参数传递给函数。`toarray()`方法用于将稀疏矩阵转换为密集矩阵。
阅读全文