save_npz
- sparse.save_npz(filename, matrix, compressed=True)[source]
Save a sparse matrix to disk in numpy’s
.npz
format. Note: This is not binary compatible with scipy’ssave_npz()
. This binary format is not currently stable. Will save a file that can only be opend with this package’sload_npz()
.- Parameters:
filename (string or file) – Either the file name (string) or an open file (file-like object) where the data will be saved. If file is a string or a Path, the
.npz
extension will be appended to the file name if it is not already therematrix (SparseArray) – The matrix to save to disk
compressed (bool) – Whether to save in compressed or uncompressed mode
Examples
Store sparse matrix to disk, and load it again:
>>> import os >>> import sparse >>> import numpy as np >>> dense_mat = np.array([[[0.0, 0.0], [0.0, 0.70677779]], [[0.0, 0.0], [0.0, 0.86522495]]]) >>> mat = sparse.COO(dense_mat) >>> mat <COO: shape=(2, 2, 2), dtype=float64, nnz=2, fill_value=0.0> >>> sparse.save_npz("mat.npz", mat) >>> loaded_mat = sparse.load_npz("mat.npz") >>> loaded_mat <COO: shape=(2, 2, 2), dtype=float64, nnz=2, fill_value=0.0> >>> os.remove("mat.npz")