U Gv f@s8ddlZddlZddgZeddZd ddZddZdS) Nsave_npzload_npzF)Z allow_pickleTcCsi}|jdkr"|j|j|jdnH|jdkr<|j|jdn.|jdkrZ|j|j|jdntd|j|j|jd|j |j d |rt j |f|nt j |f|d S) 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) ZcscZcsrZbsr)indicesindptrdia)offsetscoo)rowcolz7Save is not implemented for sparse matrix of format {}.ascii)formatshapedataN)r updaterrrr r NotImplementedErrorencoderrnpZsavez_compressedZsavez)fileZmatrix compressedZ arrays_dictr;/tmp/pip-unpacked-wheel-96ln3f52/scipy/sparse/_matrix_io.pyr s -    c Csvtj|ft\}z |d}Wn2tk rP}ztd||W5d}~XYnX|}t|tsn| d}zt t j d|}Wn2t k r}ztd||W5d}~XYnX|dkr||d|d |d f|d d W5QRS|d kr ||d|df|d d W5QRS|dkrZ||d|d|dff|d d W5QRStd|W5QRXdS)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) r z-The file {} does not contain a sparse matrix.Nr z {}_matrixzUnknown matrix format "{}"rrrrr)rrrr r r z7Load is not implemented for sparse matrix of format {}.)rload PICKLE_KWARGSKeyError ValueErrorr item isinstancestrdecodegetattrscipysparseAttributeErrorr)rZloadedZ matrix_formateclsrrrrLs*1 "  ". ( 0)T) ZnumpyrZ scipy.sparser!__all__dictrrrrrrrs   A