In [1]:
import numpy as np
from scipy.sparse import csr_matrix, csc_matrix, lil_matrix, save_npz, load_npz

In [2]:
l = [[0, 1, 2],
     [3, 0, 4],
     [0, 0, 0]]

In [3]:
csr = csr_matrix(l)
csc = csc_matrix(l)

In [4]:
save_npz('data/temp/csr.npz', csr)
save_npz('data/temp/csc.npz', csc)

In [5]:
csr_ = load_npz('data/temp/csr.npz')
print(csr_.toarray())


[[0 1 2]
 [3 0 4]
 [0 0 0]]

In [6]:
print(type(csr))


<class 'scipy.sparse.csr.csr_matrix'>

In [7]:
csc_ = load_npz('data/temp/csc.npz')
print(csc_.toarray())


[[0 1 2]
 [3 0 4]
 [0 0 0]]

In [8]:
print(type(csc))


<class 'scipy.sparse.csc.csc_matrix'>

In [9]:
lil = lil_matrix(l)

In [10]:
# save_npz('data/temp/lil.npz', lil)
# NotImplementedError: Save is not implemented for sparse matrix of format lil.

In [11]:
npz = np.load('data/temp/csr.npz')

In [12]:
print(npz.files)


['indices', 'indptr', 'format', 'shape', 'data']

In [13]:
print(npz['data'])


[1 2 3 4]

In [14]:
print(npz['indices'])


[1 2 0 2]

In [15]:
print(npz['indptr'])


[0 2 4 4]

In [16]:
print(npz['format'])


b'csr'

In [17]:
print(npz['shape'])


[3 3]