ó î&]\c@`s¥ddlmZmZmZddlZddlZddlZddl m Z ddgZ e ej ƒdkrƒe deƒZn e ƒZed„Zd „ZdS( i(tdivisiontprint_functiontabsolute_importN(t NumpyVersiontsave_npztload_npzs1.10.0t allow_picklecC`súi}|jdkr4|jd|jd|jƒnk|jdkrY|jd|jƒnF|jdkr‡|jd |jd |jƒntd j|jƒƒ‚|jd |jjd ƒd|j d|j ƒ|ræt j ||nt j ||dS(sL 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 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.todense() matrix([[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.todense() matrix([[0, 0, 3], [4, 0, 0]], dtype=int64) tcsctcsrtbsrtindicestindptrtdiatoffsetstcootrowtcols7Save is not implemented for sparse matrix of format {}.tformattasciitshapetdataN(RRR (RtupdateR R R RRtNotImplementedErrortencodeRRtnptsavez_compressedtsavez(tfiletmatrixt compressedt arrays_dict((s6lib/python2.7/site-packages/scipy/sparse/_matrix_io.pyRs,   c C`stj|tw}y|d}Wn&tk rKtdj|ƒƒ‚nX|jƒ}tjddkrt |t ƒ r|j dƒ}nyt t jdj|ƒƒ}Wn&tk rÔtdj|ƒƒ‚nX|dkr ||d |d |d fd|dƒS|dkr8||d |dfd|dƒS|dkrp||d |d|dffd|dƒStdj|ƒƒ‚WdQXdS(s´ 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 ------ IOError 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 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.todense() matrix([[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.todense() matrix([[0, 0, 3], [4, 0, 0]], dtype=int64) Rs-The file {} does not contain a sparse matrix.iiRs {}_matrixsUnknown matrix format "{}"RRR RR R RR R RRRs7Load is not implemented for sparse matrix of format {}.N(RRR (Rtloadt PICKLE_KWARGStKeyErrort ValueErrorRtitemtsyst version_infot isinstancetstrtdecodetgetattrtscipytsparsetAttributeErrorR(Rtloadedt matrix_formattcls((s6lib/python2.7/site-packages/scipy/sparse/_matrix_io.pyRSs(0  #  ) " , (t __future__RRRR$tnumpyRt scipy.sparseR*tscipy._lib._versionRt__all__t __version__tdicttFalseR tTrueRR(((s6lib/python2.7/site-packages/scipy/sparse/_matrix_io.pyts      @