_Mh^}dZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZmZgdZdZdZdd d Zdd ZGd dZdZdS)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_arrayissparse coo_matrix)mminfommreadmmwriteMMFilectt|tr|dSt|S)Nlatin1) isinstancebytesdecodestr)ss N/var/www/html/test/jupyter/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') )rinfo)sources rrrs` ;;v  rTspmatrixcHt||S)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-like object. spmatrix : bool, optional (default: True) If ``True``, return sparse ``coo_matrix``. Otherwise return ``coo_array``. Returns ------- a : ndarray or coo_array or coo_matrix Dense or sparse array 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), spmatrix=False) >>> m >>> m.toarray() 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.]]) r )rread)rr!s rrrTs^ 88==(= 3 33rcRt||||||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_array >>> 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 0 0 2.5 0 0 0 6.25 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.00e+00 0.00e+00 0.00e+00 2.50e+00 0.00e+00 0.00e+00 0.00e+00 6.25e+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_array(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.0e+00 0.0e+00 1.0e+00 -2.0e+00 4.0e+00 3.0e+00 1.0e+00 0.0e+00 0.0e+00 5.0e+00 2.5e+00 0.0e+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#d$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|jSr5)r/r6s rcolsz MMFile.colsr9rc|jSr5)r0r6s rentrieszMMFile.entriess }rc|jSr5)r1r6s rformatz MMFile.format s |rc|jSr5)r2r6s rr*z MMFile.fields {rc|jSr5)r3r6s rr,zMMFile.symmetrys ~rc8|j|j|j|jfvSr5)r3SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIANr6s r has_symmetryzMMFile.has_symmetrys'~$"9"&">"&"9";; ;r coordinatearraycP||jvrd|d|j}t|dS)Nzunknown format type , must be one of ) FORMAT_VALUES ValueError)r7r?msgs r_validate_formatzMMFile._validate_format#s< + + +VVV$BTVVCS// ! , +rintegerunsigned-integerrcomplexpatterncP||jvrd|d|j}t|dS)Nzunknown field type rJ) FIELD_VALUESrL)r7r*rMs r_validate_fieldzMMFile._validate_field2s< ) ) )SSS@QSSCS// ! * )rgeneral symmetriczskew-symmetric hermitiancL||jvrtd|d|jdS)Nzunknown symmetry type rJ)SYMMETRY_VALUESrL)r7r,s r_validate_symmetryzMMFile._validate_symmetry@sS 4/ / /FhFF/3/CFFGG G 0 /rintpuint64dDcdSr5rarrreaderz MMFile.readerM rcdSr5rararrwriterz MMFile.writerRrcrc||\}} |}d|D\}}}}} |dst d|dkst d|z|dkr|j}n|dkr|j}|rH|r1|dd vr|}nn|H| 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'. c3XK|]%}t|V&dSr5)rstrip).0parts r zMMFile.info..ys2>>tzz||$$>>>>>>rz%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: rHrGr%%zHeader line not of length 2: asciizHeader line not of length 3: )_openreadlinesplit startswithrLlower FORMAT_ARRAYFORMAT_COORDINATElstriprhlenrmapintclose)r7rstreamclose_itlinemmidrlr?r*r, split_liner8r;r=s rrz MMFile.infoWs6 ::f--/ ??$$D>>>>> 2D&&%??#344 J !HIII<<>>X-- !@4!GHHH||~~((*</// ;;==T[[]]1%5%B%B!??,,DD  jjll )((jjll )J***:!++$%D%)[[%9%9&:;;; j11 d+:!++$%D%)[[%9%9&:;;;&)#z&:&:#dG$NN$$&   x   s H>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*=dSr5)range)rrrrr(ns rrz+MMFile._get_symmetry..symm_iteratorsuq11A"1a[[11#$Q47AaDGS"Ca00000111rrFignore)over)shaperSYMMETRY_GENERALdtypecharr tocoononzerosumtodoknperrstaterrCrDrE) 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)Nra) _init_attrs)r7kwargss r__init__zMMFile.__init__0s!""6"""""rTr c>||\}} ||||}|r|n#|r|wwxYw|r$t |t rt |}|S)aB 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. spmatrix : bool, optional (default: True) If ``True``, return sparse ``coo_matrix``. Otherwise return ``coo_array``. Returns ------- a : ndarray or coo_array or coo_matrix Dense or sparse array depending on the matrix format in the Matrix Market file. )rs _parse_header _parse_bodyr~rr r)r7rr!rrdatas rr#z MMFile.read4s$ ::f--    v & & &##F++D       $ 433 $d##D 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)rs_writer~flush) r7r'r(r)r*r+r,rrs rr&z MMFile.writeUs. ::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)Nra)riattrs r z&MMFile._init_attrs..s 333TQRR333rzfound z, invalid keyword arguments, please only use rN) __class__ __slots__setkeysrLtuplesetattrr)r7rattrs public_attrs invalid_keysrs rrzMMFile._init_attrsxs (33U333 6;;==))C ,=,==  KJeL&9&9JJ;GJJKK K < |dkrt7||f|St!|d}t!|d}|rt9|d}nM| rt!|d }n9| rt!|d }n%| rt!|d }nt!|d }d}|D]}|r|ddvs|s#|dz|krt3d |}t+t&|dd\||<||<|sx| rt'|d||<n]| rt'|d||<nB| r(t)t+t,|dd||<nt-|d||<|dz }||krt3d|dz}|dz}| ro||k}||}||}||}t;||f}t;||f}| r|dz}n|r|}t;||f}t7|||ff||f|}nt?||S)N)rr)rrrrmz$Parse error, did not read all lines.intcint8r\r]rQfloatz5'entries' in header is smaller than number of entriesrpz4'entries' in header is larger than number of entries)rr) r8r;r=r?r*r,DTYPES_BY_FIELDrrFrrrrDrE FIELD_PATTERNrxrrtrhr}rQr|rrurrLryr r r conjugateNotImplementedError)r7rr8r;r=r?r*symmrrF is_integeris_unsigned_integer is_complexis_skewis_herm is_patternr(rrrrIJV entry_numberlmaskod_Iod_Jod_Vs rrzMMFile._parse_bodys48Ity48L$+48J 4O0dGVUD$((55( d00 #t'::d00 $66$11d00 T& & &tTl%000ADDAq !Q$tax<<FA '((tAw)334::<<3&d))CC(&d))CC&!3udjjll#;#;G++$&9:::JJLL36sAbqbE??0,<!6!6*-ad)),,6*-ad)),#6*13uae3D3D*E,*/!++,! g%% "5666 FA FA +QwwwD **D **,BJDD,>>++DD **1q!f+dD\GGGAA%f-- -rc|t|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'|d }|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]0} d| d} || d1|||}||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 ||&}||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 =e = =>>>''))C4000&)g&8#*=!>!$)>>r)rr$NNN)-__name__ __module__ __qualname__rpropertyr8r;r=r?r*r,rFryrxrK classmethodrNrrrrrrTrUrrCrDrErZr[r staticmethodrbrerrsrrrr#r&rrrrrarrrrsIXXXXXX;;X; %L& 5M""[" M'NJMM!>:}!#L""[" !$.$');.0BDOGG[G %f%x!3$c$c +O  \   \ KK[K\;;;\;|='='\='@##\####(,BBF    F<<<$999EEEPCGW?W?W?W?W?W?rrcg} 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)rappendGzipFile ImportErrorrrrr)rbad_clsrrs r_is_fromfile_compatiblersG  t}%%%%        s{####      GnnG&'** **s# 00A A A r)__doc__rnumpyrrrrrrrr r r scipy.sparser r r__all__rrrrrrrarrrsd ######################9888888888 3 3 3 000j $/4/4/4/4/4hlClClClC`x ?x ?x ?x ?x ?x ?x ?x ?v+++++r