xf<ddlZddlZddgZedZddZdZy) Nsave_npzload_npzF) allow_picklecLi}|jdvr(|j|j|jn||jdk(r|j|jnP|jdk(r(|j|j |j ntd|jd|j|jjd |j|j |rtj|fi|y tj|fi|y ) ac Save a sparse matrix to a file using ``.npz`` format. Parameters ---------- file : str or file-like object Either the file name (string) or an open file (file-like object) where the data will be saved. If file is a string, the ``.npz`` extension will be appended to the file name if it is not already there. matrix: spmatrix (format: ``csc``, ``csr``, ``bsr``, ``dia`` or coo``) The sparse matrix to save. compressed : bool, optional Allow compressing the file. Default: True See Also -------- scipy.sparse.load_npz: Load a sparse matrix from a file using ``.npz`` format. numpy.savez: Save several arrays into a ``.npz`` archive. numpy.savez_compressed : Save several arrays into a compressed ``.npz`` archive. Examples -------- Store sparse matrix to disk, and load it again: >>> 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 '' 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 '' with 2 stored elements in Compressed Sparse Column format> >>> sparse_matrix.toarray() array([[0, 0, 3], [4, 0, 0]], dtype=int64) csccsrbsr)indicesindptrdia)offsetscoo)rowcolz4Save is not implemented for sparse matrix of format .ascii)formatshapedataN)rupdater r rrrNotImplementedErrorencoderrnpsavez_compressedsavez)filematrix compressed arrays_dicts 9/usr/lib/python3/dist-packages/scipy/sparse/_matrix_io.pyrr sZK }}--6>>&--H % 6>>2 % vzzvzz:!$XY_YfYfXggh"ijj}}##G,ll [[  D0K0 %%ctj|fit5} |d}|j }t |ts|jd} ttj|d}|d vr"||d |d |d f|d cdddS|dk(r||d |df|d cdddS|dk(r#||d |d|dff|d cdddStdj|#t$r}t d|d|d}~wwxYw#t$r}t d|d|d}~wwxYw#1swYyxYw)a Load a sparse matrix from a file using ``.npz`` format. Parameters ---------- file : str or file-like object Either the file name (string) or an open file (file-like object) where the data will be loaded. Returns ------- result : csc_matrix, csr_matrix, bsr_matrix, dia_matrix or coo_matrix A sparse matrix containing the loaded data. Raises ------ OSError If the input file does not exist or cannot be read. See Also -------- scipy.sparse.save_npz: Save a sparse matrix to a file using ``.npz`` format. numpy.load: Load several arrays from a ``.npz`` archive. Examples -------- Store sparse matrix to disk, and load it again: >>> 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 '' 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 '' with 2 stored elements in Compressed Sparse Column format> >>> sparse_matrix.toarray() array([[0, 0, 3], [4, 0, 0]], dtype=int64) rz The file z" does not contain a sparse matrix.Nr_matrixzUnknown matrix format ""rrr r r)rr rrrrz7Load is not implemented for sparse matrix of format {}.)rload PICKLE_KWARGSKeyError ValueErroritem isinstancestrdecodegetattrscipysparseAttributeErrorrr)rloaded matrix_formateclss r!rrLsb  ' '[6 Z"8,M&**, --*009M P%,,=/(ABC 1 1vy(96(;KLTZ[bTcd'[[(e #vy(9:&/R+[[,e #vu (FGvV]_/[[2&'DDJF=DY[ [- Zy.PQRXY Y Z P6}oQGHa O P[[scD>C<1D>D1D>D>:D>"D>< DDDD> D;&D66D;;D>>E)T) numpyr scipy.sparser/__all__dictr'rrr"r!r;s. z "%( >&BK[r"