如何保存从 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)

稀疏包中 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。

How do I save a sparse matrix created from the sparse library in python