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)