hdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZgdZdZdZdZdd ZGd d Zd ZdS)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_matrix isspmatrix)mminfommreadmmwriteMMFilectt|tr|dSt|S)Nlatin1) isinstancebytesdecodestr)ss J/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/scipy/io/_mmio.pyasstrrs1!U"xx!!! q66Mc6t|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') )rinfosources rrrs` ;;v  rcDt|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.]]) )rreadrs rrrTsZ 88==  rcRt||||||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 precisionsymmetrys rrrs+X HHNN61guiBBBBBrc eZdZdZedZedZedZedZedZ edZ edZ d Z d Z e e fZed Zd Zd ZdZdZdZeeeeefZedZdZdZdZdZeeeefZedZededededediZedZ edZ!edZ"ed+dZ#ed Z$ed!Z%d"Z&d#Z' d,d&Z(d'Z)d(Z*d)Z+ d,d*Z,d%S)-r)_rows_cols_entries_format_field _symmetryc|jSN)r,selfs rrowsz MMFile.rows zrc|jSr3)r-r4s rcolsz MMFile.colsr7rc|jSr3)r.r4s rentrieszMMFile.entriess }rc|jSr3)r/r4s rformatz MMFile.format s |rc|jSr3)r0r4s rr(z MMFile.fields {rc|jSr3)r1r4s rr*zMMFile.symmetrys ~rc8|j|j|j|jfvSr3)r1SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIANr4s r has_symmetryzMMFile.has_symmetrys'~$"9"&">"&"9";; ;r coordinatearraycL||jvrtd|d|jdS)Nzunknown format type , must be one of ) FORMAT_VALUES ValueError)r5r=s r_validate_formatzMMFile._validate_format!s@ + + +*$ffd&8&8:;; ; , +rintegerunsigned-integerrcomplexpatterncL||jvrtd|d|jdS)Nzunknown field type rH) FIELD_VALUESrJ)r5r(s r_validate_fieldzMMFile._validate_field0s@ ) ) )*#eeT%6%6899 9 * )rgeneral symmetriczskew-symmetric hermitiancL||jvrtd|d|jdS)Nzunknown symmetry type rH)SYMMETRY_VALUESrJ)r5r*s r_validate_symmetryzMMFile._validate_symmetry>s@ 4/ / /*&hh(<(<>?? ? 0 /rintpuint64dDcdSr3r^rrreaderz MMFile.readerK rcdSr3r^r^rrwriterz MMFile.writerPr`rc||\}} |}d|D\}}}}} |dst d|dkst d|z|dkr|j}n|dkr|j}|r*|dd vr |}|r |dd v |s(|}|(|} ||jkrVt| d ks%t d | d ztt| \} } | | z} nQt| d ks%t d| d ztt| \} } } | | | ||| f|r| SS#|r| wwxYw)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'. cPg|]#}t|$Sr^)rstrip).0parts r zMMFile.info..ws(>>>tzz||$$>>>rz%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: rFrEr%%zHeader line not of length 2: asciizHeader line not of length 3: )_openreadlinesplit startswithrJlower FORMAT_ARRAYFORMAT_COORDINATErelenrmapintclose)r5rstreamclose_itlinemmidrir=r(r* split_liner6r9r;s rrz MMFile.infoUsw6 ::f--, ??$$D>>>>> 2D&&%??#344 J !HIII<<>>X-- !@4!GHHH||~~((*</// )47i//(( )47i//jjll )((jjll )J***:!++$%D%)[[%9%9&:;;; j11 d+:!++$%D%)[[%9%9&:;;;&)#z&:&:#dG$NN$$&   x   s HII)rbc tj|}n#t$r|dfcYSwxYw|ddkrtj|swtj|dzr|dz}nOtj|dzr|dz}n'tj|dzr|dz}|drddl}|||}nd|d rddl}| |d }n4t||}n#|d ddkr|dz}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.bz2rT) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)filespecmoderr{rs rrpz MMFile._opens4 #y**HH # # #U? " " " # 7c>>7>>(++ 57>>(6/225'&0HHW^^HY$6775')3HHW^^HZ$7885'*4H  '' . 8T22""6** . Xt44h--}&&#f,(D))Ft|s ((c j\} | kr tjSd}d}jjdv}t r\}}||k||kkr tjS fd}n fd}|D]t\}} } |r | r |dkrd}nY|r|| krd}tj d5|r || krd}dddn #1swxYwY|r|t| krd}|s|s|snu|r tj S|r tjS|r tjStjS) NTFDc3KD]-\\}}}||kr||f}||dfV ||kr||dfV.dS)NFT)items)ijaijajir&s r symm_iteratorz+MMFile._get_symmetry..symm_iteratorsy%&WWYY//MVaS1uu1g"C/////a"C.... //rc3KtD]<}t|D])}||||}}||||kfV*=dSr3)range)rrrrr&ns rrz+MMFile._get_symmetry..symm_iteratorsuq11A"1a[[11#$Q47AaDGS"Ca00000111rrFignore)over)shaperSYMMETRY_GENERALdtypecharr tocoononzerosumtodoknperrstaterrArBrC) r&missymmisskewishermrowcolrrr is_diagonalrs ` @r _get_symmetryzMMFile._get_symmetrys-w1 66* *% a== 1 AJS#c   S3YOO$5$555.. A / / / / / / 1 1 1 1 1 1(5}   #S#{ #+ ##((#cSjj"F[h///'''##++!&'''''''''''''''#cT#YY.."F f    -, ,  21 1  -, ,&&s DD! $D! c tjd|ztjdtjdtjd||fzi|dS)Nz%%.%ie z%i z%u z%%.%ie %%.%ie )r FIELD_REAL FIELD_INTEGERFIELD_UNSIGNED FIELD_COMPLEXget)r(r)s r_field_templatezMMFile._field_template!sP!: #9$f%v$&7 *'+ #eT""  #rc |jdi|dS)Nr^) _init_attrs)r5kwargss r__init__zMMFile.__init__+s!""6"""""rc||\}} |||||r|SS#|r|wwxYw)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. )rp _parse_header _parse_bodyrz)r5rr{r|s rr!z MMFile.read/s  ::f--    v & & &##F++   x   s )AA4r"Nc&||d\}} ||||||||r|dS|dS#|r|w|wxYw)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)rp_writerzflush) r5r%r&r'r(r)r*r{r|s rr$z MMFile.writeJs. ::fd33  KK7E9h G G G      s A##-Bc D|jj}d|D}t|t|z }|r"t dt |d||D]/}t ||||ddd0dS)zr Initialize each attributes with the corresponding keyword arg value or a default of None c"g|] }|dd S)Nr^)rfattrs rrhz&MMFile._init_attrs..ts 333TQRR333rzfound zL invalid keyword arguments, please only use rN) __class__ __slots__setkeysrJtuplesetattrr)r5rattrs public_attrs invalid_keysrs rrzMMFile._init_attrsms (33U333 6;;==))C ,=,==  <*-2<-@-@-@-@-9\;<< < < G++$&9:::JJLL36sAbqbE??0,<!6!6*-ad)),,6*-ad)),#6*13uae3D3D*E,*/!++,! g%% "5666 FA FA +QwwwD **D **,BJDD,>>++DD ** A1v;tTl%HHHAA%f-- -s6 AAczt|ts:t|ts%t|tst |dr|j}t |}t|jdkrtd|j\}} |||j kr:t|j dstd|d}n||jkr$|j jdvr|d}ni||jkr#|j jdvr|d }n:t%|std t'|zd }|j\}} |j j} | | d vrd }nd}|f|j j} | dkr't|j dstdd}n-| dkrd}n$| dkrd}n| dkrd}nt+d| z|||}|j||j||j|d|d|d|d} || d|dD]/} d| z} || d0|||}||jkrd|| fz} || d||j |j|jfvr,||j kr[tC| D]I}tC|D]7}||||fz} || d8JdS||j"kr_tC| D]M}tC|dz|D]7}||||fz} || d8NdStC| D]J}tC||D]7}||||fz} || d8KdS||jkr||j krytC| D]g}tC|D]U}|||f}|tG|tI|fz} || dVhdStC| D]h}tC||D]U}|||f}|tG|tI|fz} || dVidS||j%krtdt+d |z|&}||j krK|j'|j(k}tS|j*||j'||j(|ff|j!}d"|| |j+fz} || d|||dz }||j%krWtY|j'dz|j(dzD]4\}}d||fz} || d5dS||j |j|jfvrdtY|j'dz|j(dz|j*D];\}}}d#||fz||zz} || d)>????@@@ !==="4[[@@!&q1ud!3!3@@A#+a1g#5D"LLX)>)>????@@@ #4[[@@!&q$@@A#+a1g#5D"LLX)>)>????@@@ $,,,t444"4[[@@!&t@@A"#AqD'C#+tCyy$s)).D#DD"LLX)>)>????@@@ #4[[@@!&q$@@A"#AqD'C#+tCyy$s)).D#DD"LLX)>)>????@@@ $,,, !MNNN 7% ?@@@''))C4000&)g&8# #(+>"?"%'*=">"%'*=">"@"A(+y222  4sw"77D LLX.. / / /++E9Q;??H*** 37195588DAq$1v-DLLX!6!67777884-t.000"3719cgaiBB88GAq!$1v-(Q,?DLLX!6!6777788$,,,"3719cgaiBB88GAq!$1v-(afaf=M2MNDLLX!6!6777788 7% ?@@@r)rr"NNN)-__name__ __module__ __qualname__rpropertyr6r9r;r=r(r*rDrvrurI classmethodrKrrrrrrQrRrrArBrCrWrXr staticmethodr_rbrrprrrr!r$rrrrr^rrrrsIXXXXXX;;X; %L& 5M;;[; M'NJMM!>:}!#L99[9 !$.$');.0BDO??[? %f%x!3$c$c +O  \   \ HH[HV;;;\;|='='\='@##\####6BF    F<<<&999kkk\CGWAWAWAWAWAWArrcg} ddl}||jn#t$rYnwxYw ddl}||jn#t$rYnwxYwt |}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)rappendGzipFilerrrrr)r{bad_clsrrs r_is_fromfile_compatiblersG  t}%%%%        s{####      GnnG&'** **s# 00A A A r)__doc__rnumpyrrrrrrrr r r rr r __all__rrrrrrr^rrrsW ######################0/////// 3 3 3 000j-!-!-!dlClClClC`V AV AV AV AV AV AV AV Ar+++++r