稀疏矩阵的操作
以下列出一些scipy.sparse
里面对于稀疏矩阵的一些方法:
- 对于一个
sparse.spmatrix
(抽象基类),可以用.tocoo, tocsr
等方法来进行存储方式的转换。 -
todense()
方法将稀疏矩阵变成稠密矩阵,注意转换后的内存消耗。 -
sparse.save_npz
函数将稀疏矩阵序列化,sparse.load_npz
反序列化。 -
sparse.hstack
和sparse.vstack
函数用于稀疏矩阵的拼接。 - 稀疏矩阵与向量相乘用
.dot
方法。 -
scipy.sparse.linalg
包有稀疏矩阵的一些线性代数方法
a = sparse.csr_matrix([[1, 2, 3],
[2, 3, 4]])
b = sparse.csr_matrix([[8],
[9],
[10]])
# 计算两个稀疏矩阵的乘积
a.dot(b)
# a.multiply 方法用于计算 point-wise multiplication