使用python 对稀疏矩阵保存和读取

使用scipy包

import scipy.sparse as sparse
import scipy.io as sio
import scipy.stats as stats
import numpy as np

创建一个稀疏矩阵

np.random.seed(42)
rvs = stats.poisson(15, loc=10).rvs
sparse_matrix = sparse.random(500, 
                  25,
                  density=0.25,
                  format="csr")

将稀疏矩阵转换成稠密矩阵

sparse_matrix.todense()
sparse_matrix.toarray()

 

将稀疏矩阵保存为mtx格式文件

sio.mmwrite("sparse_matrix.mtx",sparse_matrix)

# 读取mtx格式文件
sp_matrix=sio.mmread("sparse_matrix.mtx")

可以通过生成mtx文件和想要读取的数据集的格式进行对比可以找到程序错误