Python的稀疏矩阵和参数保存 save/load

from scipy.sparse import coo_matrix

# 建立稀疏矩阵
data = [1,2,3,4]
row = [3,6,8,2]
col = [0,7,4,9]

c = coo_matrix((data,(row,col)),shape=(10,10)) #构建10*10的稀疏矩阵，其中不为0的值和位置在第一个参数
print(c)

d = c.todense()
print(d)

e = coo_matrix(d)  #将一个0值很多的矩阵转为稀疏矩阵
print(e)

import numpy as np

# numpy.save(arg_1,arg_2)，arg_1是文件名，arg_2是要保存的数组

aa = np.array(d)
print(aa)

# save
np.save(‘test_save_1.npy‘, aa)  #保存一个数组
np.savez(‘test_save_2‘, aa=aa, d=d)  #保存多个数组，其中稀疏矩阵可以直接保存

#load
a_ = np.load(‘test_save_1.npy‘)
print(a_)

dt = np.load(‘test_save_2.npz‘)  #npz数据加载后是一个字典格式数据
print(dt)
print(dt[‘aa‘])
print(dt[‘d‘])  #获取其中的参数值，类似于字典形式获取

#获取参数名称
p_name =list(dt.keys())
print(p_name)

#获取值
p_value =list(dt.values())
print(p_value)