python - 如何保存和加载 scipy 稀疏 csr 矩阵字典？

Question

我有一个dict的scipy.sparse.csr_matrix对象作为值，带有整数键。我怎样才能将其保存在单独的文件中？

如果我有一个常规的ndarray对于每个条目，我可以使用 json 对其进行序列化，但是当我用稀疏矩阵尝试这个时:

    with open('filename.txt', 'w') as f:
            f.write(json.dumps(the_matrix))

我收到一个类型错误:

TypeError: <75x75 sparse matrix of type '<type 'numpy.int64'>'
with 10 stored elements in Compressed Sparse Row format> is not JSON serializable

如何使用整数键和稀疏 csr 矩阵值保存字典？

Answer 1

我在尝试保存值为 csr_matrix 的字典时遇到了同样的问题。使用 pickle 将其转储到磁盘。文件处理程序应以“wb”模式打开。

import pickle
pickle.dump(csr_dict_obj, open("csr_dict.pkl","wb"))

使用加载字典。

csr_dict = pickle.load(open("csr_dict.pkl","rb"))