Python SciPy sparse.save_npz用法及代码示例

本文简要介绍 python 语言中 scipy.sparse.save_npz 的用法。

用法:

scipy.sparse.save_npz(file, matrix, compressed=True)#

使用.npz 格式将稀疏矩阵保存到文件中。

参数 ：：

file： str 或 file-like 对象

将保存数据的文件名(字符串)或打开的文件(file-like 对象)。如果文件是一个字符串，.npz 扩展名将被附加到文件名(如果它不存在)。

matrix: spmatrix (format: ``csc``, ``csr``, ``bsr``, ``dia`` or coo``)：

要保存的稀疏矩阵。

compressed： 布尔型，可选

允许压缩文件。默认值：真

例子：

将稀疏矩阵存储到磁盘，然后再次加载：

>>> import numpy as np
>>> import scipy.sparse
>>> sparse_matrix = scipy.sparse.csc_matrix(np.array([[0, 0, 3], [4, 0, 0]]))
>>> sparse_matrix
<2x3 sparse matrix of type '<class 'numpy.int64'>'
   with 2 stored elements in Compressed Sparse Column format>
>>> sparse_matrix.toarray()
array([[0, 0, 3],
       [4, 0, 0]], dtype=int64)

>>> scipy.sparse.save_npz('/tmp/sparse_matrix.npz', sparse_matrix)
>>> sparse_matrix = scipy.sparse.load_npz('/tmp/sparse_matrix.npz')

>>> sparse_matrix
<2x3 sparse matrix of type '<class 'numpy.int64'>'
   with 2 stored elements in Compressed Sparse Column format>
>>> sparse_matrix.toarray()
array([[0, 0, 3],
       [4, 0, 0]], dtype=int64)