o ?HhH@s8ddlZddlZddgZeddZd ddZddZdS) Nsave_npzload_npzF) allow_pickleTcCsi}|jdvr|j|j|jdn'|jdkr|j|jdn|jdkr-|j|j|jdn d|jd}t||j|jd |j |j d t |t j jrS|jd d |r`tj|fi|d Stj|fi|d S)aj Save a sparse matrix or array 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 or sparray The sparse matrix or array to save. Supported formats: ``csc``, ``csr``, ``bsr``, ``dia`` or ``coo``. 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 as sp >>> sparse_matrix = sp.sparse.csc_matrix([[0, 0, 3], [4, 0, 0]]) >>> sparse_matrix >>> sparse_matrix.toarray() array([[0, 0, 3], [4, 0, 0]], dtype=int64) >>> sp.sparse.save_npz('/tmp/sparse_matrix.npz', sparse_matrix) >>> sparse_matrix = sp.sparse.load_npz('/tmp/sparse_matrix.npz') >>> sparse_matrix >>> 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)formatshapedataT) _is_arrayN)rupdater r r rrNotImplementedErrorencoderr isinstancespsparsesparraynpsavez_compressedsavez)filematrix compressed arrays_dictmsgr%W/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/sparse/_matrix_io.pyr s&.     c Csttj|fit}|d}|durtd|d|}t|ts)|d}|dr3|d}n|d}z t t j |}Wnt yU}z td |d |d}~ww|d vrr||d |d |df|ddWdS|dkr||d |df|ddWdS|dkr||d |d|dff|ddWdSt d|d1swYdS)a Load a sparse array/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_array, csr_array, bsr_array, dia_array or coo_array A sparse array/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 array/matrix to a file using ``.npz`` format. numpy.load: Load several arrays from a ``.npz`` archive. Examples -------- Store sparse array/matrix to disk, and load it again: >>> import numpy as np >>> import scipy as sp >>> sparse_array = sp.sparse.csc_array([[0, 0, 3], [4, 0, 0]]) >>> sparse_array >>> sparse_array.toarray() array([[0, 0, 3], [4, 0, 0]], dtype=int64) >>> sp.sparse.save_npz('/tmp/sparse_array.npz', sparse_array) >>> sparse_array = sp.sparse.load_npz('/tmp/sparse_array.npz') >>> sparse_array >>> sparse_array.toarray() array([[0, 0, 3], [4, 0, 0]], dtype=int64) In this example we force the result to be csr_array from csr_matrix >>> sparse_matrix = sp.sparse.csc_matrix([[0, 0, 3], [4, 0, 0]]) >>> sp.sparse.save_npz('/tmp/sparse_matrix.npz', sparse_matrix) >>> tmp = sp.sparse.load_npz('/tmp/sparse_matrix.npz') >>> sparse_array = sp.sparse.csr_array(tmp) rNz The file z+ does not contain a sparse array or matrix.rr_array_matrixzUnknown format ""rrr r r)rr r r rrz4Load is not implemented for sparse matrix of format r)rload PICKLE_KWARGSget ValueErroritemrstrdecodegetattrrrAttributeErrorr)r loaded sparse_format sparse_typeclser%r%r&rPsF6       )T) numpyrscipyr__all__dictr+rrr%r%r%r&s    E