如何在python中保存从稀疏库创建的稀疏矩阵

时间:2019-08-13 13:44:25

标签: python numpy scipy sparse-matrix

我使用python中的稀疏库创建了一个5维稀疏矩阵,我的代码看起来与文档类似:​​

import sparse
coords = [[0, 1, 2, 3, 4],
           [0, 1, 2, 3, 4]]
data = [10, 20, 30, 40, 50]
s = sparse.COO(coords, data, shape=(5, 5))

我想保存这个矩阵,以便以后可以用类似的东西加载它:

sp.save_npz("filename.npz", s)

稀疏包中的save_npz(在scipy中)等效功能是什么?

1 个答案:

答案 0 :(得分:3)

与numpy或scipy中使用的命令相同:

sparse.save_npz('filename.npz', s, compressed=True)

它实际上以numpy的npz格式保存它,但这包含在稀疏API中。

相应地,还有一个sparse.load_npz()

Sparse与scipy和numpy集成非常容易,实际上,您甚至可以在numpy或scipy与稀疏矩阵之间进行转换。

签出here