Python稀疏矩阵和普通矩阵做乘法代码
时间: 2024-02-17 17:03:00 浏览: 140
python稀疏矩阵乘法
3星 · 编辑精心推荐
以下是 Python 中稀疏矩阵和普通矩阵做乘法的示例代码:
普通矩阵乘法:
```
import numpy as np
# 定义两个矩阵
A = np.array([[1, 2], [3, 4]])
B = np.array([[5, 6], [7, 8]])
# 普通矩阵乘法
C = np.dot(A, B)
print(C)
```
稀疏矩阵乘法:
```
import numpy as np
from scipy.sparse import csr_matrix
# 定义两个稀疏矩阵
A = csr_matrix([[1, 0], [0, 2]])
B = csr_matrix([[3, 0], [0, 4]])
# 稀疏矩阵乘法
C = A.dot(B)
print(C.toarray())
```
需要注意的是,稀疏矩阵的定义需要使用 `scipy.sparse` 库中的 `csr_matrix` 函数,乘法运算使用 `dot` 方法,最后需要用 `toarray` 方法将稀疏矩阵转换为普通矩阵。
阅读全文