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

Question

我使用 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)

sparse 包中与 save_npz(在 scipy 中)等效的函数是什么？

Answer 1

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

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

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

相应的，还有一个sparse.load_npz()。

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

检查一下 here .