a jA@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZgdZddZdd Zd d Zdd dZGdddZddZdS)z Matrix Market I/O in Python. See http://math.nist.gov/MatrixMarket/formats.html for information about the Matrix Market format. N) asarrayrealimagconjzerosndarray concatenateonescan_cast) coo_matrixissparse)mminfommreadmmwriteMMFilecCst|tr|dSt|S)Nlatin1) isinstancebytesdecodestr)sr4E:\soccer\ml_env\Lib\site-packages\scipy/io/_mmio.pyasstrs  rcCs t|S)a Return size and storage parameters from Matrix Market file-like 'source'. Parameters ---------- source : str or file-like Matrix Market filename (extension .mtx) or open file-like object Returns ------- rows : int Number of matrix rows. cols : int Number of matrix columns. entries : int Number of non-zero entries of a sparse matrix or rows*cols for a dense matrix. format : str Either 'coordinate' or 'array'. field : str Either 'real', 'complex', 'pattern', or 'integer'. symmetry : str Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. Examples -------- >>> from io import StringIO >>> from scipy.io import mminfo >>> text = '''%%MatrixMarket matrix coordinate real general ... 5 5 7 ... 2 3 1.0 ... 3 4 2.0 ... 3 5 3.0 ... 4 1 4.0 ... 4 2 5.0 ... 4 3 6.0 ... 4 4 7.0 ... ''' ``mminfo(source)`` returns the number of rows, number of columns, format, field type and symmetry attribute of the source file. >>> mminfo(StringIO(text)) (5, 5, 7, 'coordinate', 'real', 'general') )rinfosourcerrrr s0r cCs t|S)aG Reads the contents of a Matrix Market file-like 'source' into a matrix. Parameters ---------- source : str or file-like Matrix Market filename (extensions .mtx, .mtz.gz) or open file-like object. Returns ------- a : ndarray or coo_matrix Dense or sparse matrix depending on the matrix format in the Matrix Market file. Examples -------- >>> from io import StringIO >>> from scipy.io import mmread >>> text = '''%%MatrixMarket matrix coordinate real general ... 5 5 7 ... 2 3 1.0 ... 3 4 2.0 ... 3 5 3.0 ... 4 1 4.0 ... 4 2 5.0 ... 4 3 6.0 ... 4 4 7.0 ... ''' ``mmread(source)`` returns the data as sparse matrix in COO format. >>> m = mmread(StringIO(text)) >>> m <5x5 sparse matrix of type '' with 7 stored elements in COOrdinate format> >>> m.A array([[0., 0., 0., 0., 0.], [0., 0., 1., 0., 0.], [0., 0., 0., 2., 3.], [4., 5., 6., 7., 0.], [0., 0., 0., 0., 0.]]) )rreadrrrrrTs-rcCst||||||dS)a Writes the sparse or dense array `a` to Matrix Market file-like `target`. Parameters ---------- target : str or file-like Matrix Market filename (extension .mtx) or open file-like object. a : array like Sparse or dense 2-D array. comment : str, optional Comments to be prepended to the Matrix Market file. field : None or str, optional Either 'real', 'complex', 'pattern', or 'integer'. precision : None or int, optional Number of digits to display for real or complex values. symmetry : None or str, optional Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. If symmetry is None the symmetry type of 'a' is determined by its values. Returns ------- None Examples -------- >>> from io import BytesIO >>> import numpy as np >>> from scipy.sparse import coo_matrix >>> from scipy.io import mmwrite Write a small NumPy array to a matrix market file. The file will be written in the ``'array'`` format. >>> a = np.array([[1.0, 0, 0, 0], [0, 2.5, 0, 6.25]]) >>> target = BytesIO() >>> mmwrite(target, a) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix array real general % 2 4 1.0000000000000000e+00 0.0000000000000000e+00 0.0000000000000000e+00 2.5000000000000000e+00 0.0000000000000000e+00 0.0000000000000000e+00 0.0000000000000000e+00 6.2500000000000000e+00 Add a comment to the output file, and set the precision to 3. >>> target = BytesIO() >>> mmwrite(target, a, comment='\n Some test data.\n', precision=3) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix array real general % % Some test data. % 2 4 1.000e+00 0.000e+00 0.000e+00 2.500e+00 0.000e+00 0.000e+00 0.000e+00 6.250e+00 Convert to a sparse matrix before calling ``mmwrite``. This will result in the output format being ``'coordinate'`` rather than ``'array'``. >>> target = BytesIO() >>> mmwrite(target, coo_matrix(a), precision=3) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix coordinate real general % 2 4 3 1 1 1.00e+00 2 2 2.50e+00 2 4 6.25e+00 Write a complex Hermitian array to a matrix market file. Note that only six values are actually written to the file; the other values are implied by the symmetry. >>> z = np.array([[3, 1+2j, 4-3j], [1-2j, 1, -5j], [4+3j, 5j, 2.5]]) >>> z array([[ 3. +0.j, 1. +2.j, 4. -3.j], [ 1. -2.j, 1. +0.j, -0. -5.j], [ 4. +3.j, 0. +5.j, 2.5+0.j]]) >>> target = BytesIO() >>> mmwrite(target, z, precision=2) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix array complex hermitian % 3 3 3.00e+00 0.00e+00 1.00e+00 -2.00e+00 4.00e+00 3.00e+00 1.00e+00 0.00e+00 0.00e+00 5.00e+00 2.50e+00 0.00e+00 N)rwrite)targetacommentfield precisionsymmetryrrrrslrc @steZdZdZeddZeddZeddZedd Zed d Z ed d Z eddZ dZ dZ e e fZeddZdZdZdZdZdZeeeeefZeddZdZdZdZdZeeeefZedd Zed!ed"ed#ed$ed#iZed%d&Z ed'd(Z!ed)d*Z"edBd,d-Z#ed.d/Z$ed0d1Z%d2d3Z&d4d5Z'dCd8d9Z(d:d;Z)dd?Z+dDd@dAZ,d7S)Er)_rows_cols_entries_format_field _symmetrycCs|jSN)r&selfrrrrowssz MMFile.rowscCs|jSr,)r'r-rrrcolssz MMFile.colscCs|jSr,)r(r-rrrentriesszMMFile.entriescCs|jSr,)r)r-rrrformat sz MMFile.formatcCs|jSr,)r*r-rrrr#sz MMFile.fieldcCs|jSr,)r+r-rrrr%szMMFile.symmetrycCs|j|j|j|jfvSr,)r+SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIANr-rrr has_symmetryszMMFile.has_symmetry coordinatearraycCs(||jvr$d|d|j}t|dS)Nzunknown format type , must be one of ) FORMAT_VALUES ValueError)r.r2msgrrr_validate_format!s zMMFile._validate_formatintegerunsigned-integerrcomplexpatterncCs(||jvr$d|d|j}t|dS)Nzunknown field type r9) FIELD_VALUESr;)r.r#r<rrr_validate_field0s zMMFile._validate_fieldZgeneralZ symmetriczskew-symmetricZ hermitiancCs$||jvr td|d|jdS)Nzunknown symmetry type r9)SYMMETRY_VALUESr;)r.r%rrr_validate_symmetry>s  zMMFile._validate_symmetryintpuint64dDcCsdSr,rrrrrreaderKsz MMFile.readercCsdSr,rrrrrwriterPsz MMFile.writercCsz||\}}zT|}dd|D\}}}}} |dsHtd|dks`td||dkrt|j}n|dkr|j}|r|r|d d vr|}qqq| s|}q|} ||jkr t | d kstd | d t t | \} } | | } n0t | dks,td| d t t | \} } } | | | ||| fW|rb|Sn|rt|0dS)a Return size, storage parameters from Matrix Market file-like 'source'. Parameters ---------- source : str or file-like Matrix Market filename (extension .mtx) or open file-like object Returns ------- rows : int Number of matrix rows. cols : int Number of matrix columns. entries : int Number of non-zero entries of a sparse matrix or rows*cols for a dense matrix. format : str Either 'coordinate' or 'array'. field : str Either 'real', 'complex', 'pattern', or 'integer'. symmetry : str Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. css|]}t|VqdSr,)rstrip).0partrrr wzMMFile.info..z%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: r8r7r%%zHeader line not of length 2: asciizHeader line not of length 3: N)_openreadlinesplit startswithr;lower FORMAT_ARRAYFORMAT_COORDINATElstriprLlenrmapintclose)r.rstreamclose_itlineZmmidrQr2r#r%Z split_liner/r0r1rrrrUsT           z MMFile.inforbcCszt|}Wnty(|dfYS0|ddkrtj|stj|dr\|d}n2tj|drv|d}ntj|dr|d}|drddl}|||}n*|d rddl}| |d }n t||}n"|d ddkr|d}t||}|d fS) a Return an open file stream for reading based on source. If source is a file name, open it (after trying to find it with mtx and gzipped mtx extensions). Otherwise, just return source. Parameters ---------- filespec : str or file-like String giving file name or file-like object mode : str, optional Mode with which to open file, if `filespec` is a file name. Returns ------- fobj : file-like Open file-like object. close_it : bool True if the calling function should close this file when done, false otherwise. Frrz.mtxz.mtx.gzz.mtx.bz2z.gzNz.bz2rgT) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)ZfilespecmoderprdrrrrrrXs.         z MMFile._openc sTj\}|krtjSd}d}jjdv}tr|\}}||k||kkrftjS fdd}nfdd}|D]\}} } |r| r|dkrd}nf|r|| krd}t j dd "|r|| krd}Wdn1s0Y|r|t | krd}|s|s|sq*q|r6tj S|rBtjS|rNtjStjS) NTFDc3sPD]B\\}}}||kr6||f}||dfVq||kr||dfVqdS)NFT)items)ijaijaji)r!rr symm_iterators  z+MMFile._get_symmetry..symm_iteratorc3sLtD]>}t|D].}||||}}||||kfVqqdSr,)range)rxrwryrzr!nrrr{s rFignore)Zover)shaperSYMMETRY_GENERALdtypecharr tocooZnonzerosumZtodoknpZerrstaterr3r4r5) r!mZissymmZisskewZishermrowcolr{ryrzZ is_diagonalrr}r _get_symmetrysB     " zMMFile._get_symmetryc Cs0tjd|tjdtjdtjd||fi|dS)Nz%%.%ie z%i z%u z%%.%ie %%.%ie )r FIELD_REAL FIELD_INTEGERFIELD_UNSIGNED FIELD_COMPLEXget)r#r$rrr_field_template$s zMMFile._field_templatecKs|jfi|dSr,) _init_attrs)r.kwargsrrr__init__.szMMFile.__init__cCsF||\}}z$||||W|r0|Sn|r@|0dS)a Reads the contents of a Matrix Market file-like 'source' into a matrix. Parameters ---------- source : str or file-like Matrix Market filename (extensions .mtx, .mtz.gz) or open file object. Returns ------- a : ndarray or coo_matrix Dense or sparse matrix depending on the matrix format in the Matrix Market file. N)rX _parse_header _parse_bodyrc)r.rrdrerrrr2s  z MMFile.readrNc Cs\||d\}}z.|||||||W|r6|qX|n|rN|n|0dS)a7 Writes sparse or dense array `a` to Matrix Market file-like `target`. Parameters ---------- target : str or file-like Matrix Market filename (extension .mtx) or open file-like object. a : array like Sparse or dense 2-D array. comment : str, optional Comments to be prepended to the Matrix Market file. field : None or str, optional Either 'real', 'complex', 'pattern', or 'integer'. precision : None or int, optional Number of digits to display for real or complex values. symmetry : None or str, optional Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. If symmetry is None the symmetry type of 'a' is determined by its values. wbN)rX_writercflush) r.r r!r"r#r$r%rdrerrrrMs   z MMFile.writec Ksl|jj}dd|D}t|t|}|rBtdt|||D] }t||||dddqFdS)zr Initialize each attributes with the corresponding keyword arg value or a default of None cSsg|]}|ddqS)Nr)rMattrrrr wrPz&MMFile._init_attrs..zVfound {} invalid keyword arguments, please only use {}rN) __class__ __slots__setkeysr;r2tuplesetattrr)r.rattrsZ public_attrsZ invalid_keysrrrrrpszMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r/r0r1r2r#r%)rrr)r.rdr/r0r1r2r#r%rrrrs   zMMFile._parse_headercCsR|j|j|j|j|j|jf\}}}}}}|j|d}|j} ||j k} ||j k} ||j k} ||j k} ||j k}||jk}||jkr(t||f|d}d}d\}}| rd|||f<||dkr|d7}|r|}|r|ddvs|sq| rt|}n2| r t|}n"| r$ttt|}nt|}||||f<| r||kr| r^| |||f<n$|rvt||||f<n ||||f<||dkr|d}q|d}| sd}q|}| rd|||f<||dkr|d7}q| r|d|fvr||dks$tdn |d|fvr||ksNtdn&||jkrF|dkrNt||f|dSt|dd}t|dd}|rzt|dd}nH| rt|d d}n4| rt|d d}n | rt|d d}n t|d d}d}|D]}|r|ddvs|sq|d|krtd |}tt|dd\||<||<|s| rJt|d||<nJ| rbt|d||<n2| rttt|dd||<nt|d||<|d7}q||krtd|d8}|d8}| r*||k}||}||}||}t||f}t||f}| r|d9}n|r|}t||f}t|||ff||f|d}nt||S)N)rr)rrrrRz$Parse error, did not read all lines.ZintcZint8rFrGr@floatz5'entries' in header is smaller than number of entriesrUz4'entries' in header is larger than number of entries)rr) r/r0r1r2r#r%DTYPES_BY_FIELDrr6rrrr4r5 FIELD_PATTERNr]rrYrLrbr@rarrZrr;r^r r r conjugateNotImplementedError)r.rdr/r0r1r2r#Zsymmrr6 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternr!rfrwrxryIJVZ entry_numberlmaskZod_IZod_JZod_Vrrrrs                              zMMFile._parse_bodycCst|ts(t|ts(t|ts(t|dr|j}t|}t|jdkrLt d|j\}} |dur||j krt |j ds|t d|d}q||jkr|j jdvr|d}q||jkr|j jdvr|d }n&t|st d t|d }|j\}} |j j} |dur| d vrd }nd}|dur|j j} | dkrPt |j dsJt dd}n<| dkr`d}n,| dkrpd}n| dkrd}n td| |dur||}|j||j||j|d|d|d|d} || d|dD]} d| } || dq|||}||jkr*d|| f} || d||j |j|jfvrF||j krt!| D]6}t!|D]&}||||f} || dqqtn||j"krt!| D]<}t!|d|D]&}||||f} || dqԐqnBt!| D]8}t!||D]&}||||f} || dqq n||jkr||j krt!| D]F}t!|D]6}|||f}|t#|t$|f} || dqrqfnRt!| D]H}t!||D]6}|||f}|t#|t$|f} || dqƐqn"||j%krt dn td |n|&}||j krr|j'|j(k}t)|j*||j'||j(|ff|jd!}d"|| |j+f} || d|||d}||j%krt,|j'd|j(dD]&\}}d||f} || dqn||j |j|jfvrRt,|j'd|j(d|j*D]0\}}}d#||f||} || dqnn||jkrt,|j'd|j(d|j*D]8\}}}d#||f||j#|j$f} || dqxn td |dS)$NZ __array__rUzExpected 2 dimensional arrayrFzBmmwrite does not support integer dtypes larger than native 'intp'.fdrHrurIzunknown matrix type: %sr7ZfFrwr>frcr@ur?zunexpected dtype kind z%%MatrixMarket matrix   rz%%%s z%i %i rz*pattern type inconsisted with dense formatzUnknown field type %s)rz %i %i %i z%i %i )-rlistrrhasattrr]rr`rr;rr r OverflowErrorZastyperrrr typekindrlrrr=rCrErencoderZrrrr|r4rrrrrrr dataZnnzzip)r.rdr!r"r#r$r%repr/r0typecoderrrftemplaterxrwryZcooZlower_triangle_maskrhrrHrrrrs                                                 $ $z MMFile._write)rg)rNNN)rNNN)-__name__ __module__ __qualname__rpropertyr/r0r1r2r#r%r6r^r]r: classmethodr=rrrrrrBrCrr3r4r5rDrEr staticmethodrJrKrrXrrrrrrrrrrrrrrs             N > @  # rcCspg}zddl}||jWnty.Yn0zddl}||jWntyZYn0t|}t|| S)z Check whether `stream` is compatible with numpy.fromfile. Passing a gzipped file object to ``fromfile/fromstring`` doesn't work with Python 3. rN)rpappendGzipFile ImportErrorrrrsrr)rdZbad_clsrprrrrr_is_fromfile_compatibles  r)rNNN)__doc__rjnumpyrrrrrrrrr r Z scipy.sparser r __all__rr rrrrrrrrs  ,52 p;