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

使用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")
将稀疏矩阵保存为npz格式文件
sparse.save_npz('sparse_matrix.npz', sparse_matrix)

# 读取npz文件
sparse_matrix = sparse.load_npz('sparse_matrix.npz')

参考链接:https://cmdlinetips.com/2021/04/save-sparse-matrix-in-python-to-mtx-and-npz-file/

你可能感兴趣的:(python,矩阵,numpy)