^jXdZdZgdZddlmZddlZddlmZddl m Z m Z m Z dd l mZmZmZdd lmZmZmZmZmZdd lmZGd d e eZddZdZGddee ZGddeeZy)z"List of Lists sparse matrix class zrestructuredtext en) lil_array lil_matrixisspmatrix_lil) bisect_leftN)spmatrix)_spbasesparrayissparse) IndexMixin INT_TYPES_broadcast_arrays)getdtypeisshape isscalarlike upcast_scalar check_shape) _csparsetoolsceZdZdZd)dddZdZdZdZd Zd*d Z e jje _ d*d Z e jje _ d Z d ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ d Z!e jBje!_ d!dd"d#Z"e jDje"_ d$Z#e jFje#_ d+d%Z$e jHje$_ d,d&Z%e jJje%_ d-d'Z&e jLje&_ d-d(Z'e jNje'_ y). _lil_baselilNFmaxprintctj|||t||t|_t |r|j dk(r|r|j}n|j}|t|}|j|d}t|j|_ |j|_|j|_ |j|_yt|t rt#|r| t%d|\}} t|| f|_ t'j(|ft*|_ t'j(|ft*|_t-|D] } g|j| <g|j| <"yt/d |j1|}t|t2r(|j4d k7rt%d |j4d |j7||j}t|j|_ t|j|_|j|_ |j|_y#t.$r} t/d | d} ~ wwxYw) Nr)defaultrFcopyzinvalid use of shape parameterdtypez(unrecognized lil_array constructor usagezunsupported matrix typezLIL arrays don't support zD input. Use 2D)r __init__rfloatrr formatrtolilastypershape_shaperowsdata isinstancetupler ValueErrornpemptyobjectrange TypeError _ascontainerr ndim_csr_container) selfarg1r&rrrAnewdtypeMNies J/opt/ringagent/.cad-venv/lib/python3.12/site-packages/scipy/sparse/_lil.pyr!z_lil_base.__init__sth7eT59  D>{{e#IIKJJL #E?HHXEH2%agg.DKDJDIDI U #t}$$%EFF1)1a&1 HHaT8 HHaT8 q&A#%DIIaL#%DIIaL& JKK B%%d+$(QVVq[ #G >? ? !8 AID 19((4::a=8Cyy CA  J QY88CgS\C277S S12 2?Ds C&2C+cL|td|jDS|dkr|dz }|dk(rtj|jdtj }t |j|jD]6\}}t ||Dcgc] \}}|dk7s |}}}||xxdz cc<8|S|dk(rPtj|jDcgc]}tj|c}tj Stdcc}}wcc}w)Nc3FK|]}tj|ywr?)r- count_nonzero).0rWs r= z*_lil_base.count_nonzero..rsJWr''0Js!rr rrrP) rQr)r-rSr&rTzipr(rUr]r,) r5rVrXrYr)cdmaskrWs r=r]z_lil_base.count_nonzerops <J JJ J !8 AID 19((4::a=8C DII6  T&)#tn?daQ??D Q  J QY88:>))Dw!!'*DBGG 12 2@ Es D#DD!c|jd|jdf|j}|j||jd<|j||jd<|S)zReturns a view of the 'i'th row (without copying). Parameters ---------- i : int Row to return view of. Returns ------- lil_array or lil_matrix A view of the 'i'th row. rrr)_lil_containerr&rr(r))r5r;news r= getrowviewz_lil_base.getrowviewsY!!1djjm"4DJJ!Giil iil  rEc|j\}}|dkr||z }|dks||k\r td|jd|f|j}|j|dd|jd<|j |dd|j d<|S)zReturns a copy of the 'i'th row. Parameters ---------- i : int Row to return a copy of. Returns ------- lil_array or lil_matrix A copy of the 'i'th row. rzrow index out of boundsrrN)r& IndexErrorrerr(r))r5r;r9r:rfs r=getrowz_lil_base.getrowszz1 q5 FA q5AF67 7!!1a& !;iil1o iil1o  rEct|trCt|dk(r5t|dtr"t|dtr|j|St j ||S)Nr rr)r*r+rRr _get_intXintr __getitem__)r5keys r=rmz_lil_base.__getitem__sX sE "s3x1}3q69-3q69-$4$$c* *%%dC00rEctj|jd|jd|j|j||}|j j |SNrr)rlil_get1r&r(r)rtype)r5rYcolvs r=rlz_lil_base._get_intXintsJ  " "4::a=$**Q-#'99c3 8zzq!!rEct|j|jd}|j|t ||dzSrp)r0indicesr&_get_row_rangesslicer5rYrss r=_get_sliceXintz_lil_base._get_sliceXints;S[[A/0##CsCE):;;rEc|j|jt||dz}|jdkDr|j |j S|S)Nr)rwravelrxr3reshaper&)r5rYrsress r=_get_arrayXintz_lil_base._get_arrayXintsH""399;c3q50AB 88a<;;syy) ) rEc(|j|f|Sr?rwrys r=_get_intXslicez_lil_base._get_intXslices##SFC00rEcnt|j|jd}|j||SNr)r0rvr&rwrys r=_get_sliceXslicez_lil_base._get_sliceXslices0S[[A/0##C--rEc&|j||Sr?rrys r=_get_arrayXslicez_lil_base._get_arrayXslices##C--rEcjtj||jd}|j||S)Nr)rndmin)r-rUr_get_columnXarrayrys r=_get_intXarrayz_lil_base._get_intXarrays+hhs#))15%%c3//rEctj|j|jd}|j ||Sr)r-arangervr&rrys r=_get_sliceXarrayz_lil_base._get_sliceXarrays4iiTZZ]34%%c3//rEcRt|dddf|\}}|j||Sr?)r_get_arrayXarrayrys r=rz_lil_base._get_columnXarrays-$S4[#6S$$S#..rEc ^ttjt||\}}|j |j |j }tj|j d|j d|j|j|j|j|||SNrrr) mapr- atleast_2d_prepare_index_for_memoryviewrer&rr lil_fancy_getr(r))r5rYrsr;jrfs r=rz_lil_base._get_arrayXarrays2=="?S"IJ1!!!''!<##DJJqM4::a=$(IItyy$'HHchh$%q * rEc |j|jd\}}}t|||}t|}|j t||f|j }t j|jd|jd|j|j|j|j||||| |S)a Fast path for indexing in the case where column index is slice. This gains performance improvement over brute force by more efficient skipping of zeros, by accessing the elements column-wise in order. Parameters ---------- rows : sequence or range Rows indexed. If range, must be within valid bounds. col_slice : slice Columns indexed rrr) rvr&r0rRrerrlil_get_row_rangesr(r)) r5r( col_slicej_startj_stopj_stride col_rangenjrfs r=rwz_lil_base._get_row_rangess %.$5$5djjm$D!'684 ^!!3t9b/!D((A 1 )-DII),388)-)0&(B  H  rEc tj|jd|jd|j|j|||yrp)r lil_insertr&r(r))r5rYrsxs r= _set_intXintz_lil_base._set_intXints7  A 1 tyy!%Ca 9rEc ttjt|||\}}}t j |j d|j d|j|j|||yrp) rr-rrr lil_fancy_setr&r(r))r5rYrsrr;rs r=_set_arrayXarrayz_lil_base._set_arrayXarray sXbmm%B3Q%OP1a##DJJqM4::a=$(IItyy$%q! -rEctj|j|j}t ||\}}|j |||y)Nr)r-asarraytoarrayrrr)r5rYrsr_s r=_set_arrayXarray_sparsez!_lil_base._set_arrayXarray_sparses> JJqyy{$** 5 C(1 c3*rEct|trat|dk(rR|\}}t|trt|trt |r|j }t|t jr|j}|jj|}|jdkDr td|j|||St|trt|tr|tdk(rr|tdk(rdt |rY|j|jk(r@|j!||j}|j"|_|j$|_yt'j(|||y)Nr rz&Trying to assign a sequence to an itemr)r*r+rRr r rr-ndarrayitemrrrsizer,rrxr&rer(r)r __setitem__)r5rnrrYrss r=rz_lil_base.__setitem__s c5 !c#h!mHC#y)ji.HA; Aa,AJJOOA&66A:$%MNN((c1553&:c5+A5;&3%++=QKAGGtzz$9'''<FF FF tS!,rEc4|dk(r)|j|j|j}|St|j|}|j |}t |j D]%\}}|Dcgc]}||z c}|j |<'|Scc}w)Nrr)rer&rrr% enumerater))r5rBrf res_dtyperrWvals r= _mul_scalarz_lil_base._mul_scalar-s A:%%djj %CC &djj%8I++i(C'1 = 74;t|j|jD]\}}t||} || d=|| d=||_ yr?) rr&r(r)r-resizer0r`rr') r5r&new_Mnew_Nr9r:r;rYr)truncs r=rz_lil_base.resizeosE" uzz1 19 &5)DI &5)DI QY $))U3DI $))U3DI1e_ "! ! ! !  " 19 DII6 ! T#C/KL !  rEc|j||}t|jD]/\}}t|D]\}}|j|||||f<1|Sr?)_process_toarray_argsrr(r))r5rrXrbr;rYposrs r=rz_lil_base.toarraysh  & &uc 2 * ,FAs#C. ,Q))A,s+!Q$ , ,rEch|j|j|djdS)NrF)axesr)r transposer$)r5rrs r=rz_lil_base.transposes2zztz$..Du.EKKQVKWWrEc*|r|jS|Sr?r)r5rs r=r$z_lil_base.tolils 99; KrEcd|j\}}|dk(s|dk(r|j||f|jS||ztjtj j krotj }tj|dz|}d|d<tj|j|ddtj|||d}n|j|}tj||}tj|j||jtj}|jt ||}tj|dz|}d|d<tj|||ddtj||}tj||j} tj|j|tj|j | |j| ||f|jS) Nrrr)rX)maxval)rrX)r&)r&r4rr-iinfoint32maxr.rlil_get_lengthsr(cumsum_get_index_dtyperQint64lil_flatten_to_arrayr)) r5rr9r: idx_dtypeindptrnnzlengthsrvr)s r=rz_lil_base.tocsrszz1 6Q!V&&1vTZZ&@ @ Q3"((288$(( (IXXa!e95FF1I  ) )$))VABZ @ IIf& )*C--Q-7Ihhq 2G  ) )$))W =++BHH+-C--SC[-AIXXa!e95FF1I IIgYF12J ?((3i0xx4::.**499g>**499d;""D'6#:$**"MMrE)NNFr?)NN)NF)F)(__name__ __module__ __qualname___formatr!rCrGrKrNrZr __doc__r]rgrjrmrlrzrrrrrrrrrwrrrrrrrr}rrrr$rr@rEr=rrsrG-T-^"" 3oo--GO3&$1199M$.1" < 1..00/ :9- + -0  (<<''DL$'e:oo--GO.^^++FNoo--GOX ))11I MM))EMN@MM))EMrErc|j|jkDr|j|j}n4|j|jkr|j|j}|jjr,|jtj tj fvr|jtj}|jjr,|jtj tj fvr|jtj}|+|jjs|j}|||fS||fS)a Convert index and data arrays to form suitable for passing to the Cython fancy getset routines. The conversions are necessary since to (i) ensure the integer index arrays are in one of the accepted types, and (ii) to ensure the arrays are writable so that Cython memoryview support doesn't choke on them. Parameters ---------- i, j Index arrays x : optional Data arrays Returns ------- i, j, x Re-formatted arrays (x is omitted, if input was None) ) rr%flags writeabler-rrrTr)r;rrs r=rrs. ww HHQWW  177  HHQWW  77  "((/C C HHRWW  77  "((/C C HHRWW }ww  A!Qw!t rEc"t|tS)aIs `x` of lil_matrix type? .. warning:: SciPy sparse is shifting from a sparse matrix interface to a sparse array interface. In the next few releases we expect to deprecate the sparse matrix interface. For documentation of the matrix interface, see the :ref:`spmatrix interface docs `. For guidance on converting existing code to sparse arrays, see :ref:`Migration from spmatrix to sparray `. Parameters ---------- x object to check for being a lil matrix Returns ------- bool True if `x` is a lil matrix, False otherwise Examples -------- >>> from scipy.sparse import lil_array, lil_matrix, coo_matrix, isspmatrix_lil >>> isspmatrix_lil(lil_matrix([[5]])) True >>> isspmatrix_lil(lil_array([[5]])) False >>> isspmatrix_lil(coo_matrix([[5]])) False )r*r)rs r=rrs@ a $$rEceZdZdZy)ra$ Row-based LIst of Lists sparse array. This is a structure for constructing sparse arrays incrementally. Note that inserting a single item can take linear time in the worst case; to construct the array efficiently, make sure the items are pre-sorted by index, per row. This can be instantiated in several ways: lil_array(D) where D is a 2-D ndarray lil_array(S) with another sparse array or matrix S (equivalent to S.tolil()) lil_array((M, N), [dtype]) to construct an empty array with shape (M, N) dtype is optional, defaulting to dtype='d'. Attributes ---------- data : ndarray LIL format data array of the array rows : ndarray LIL format row index array of the array dtype : dtype Data type of the array shape : 2-tuple Shape of the array ndim : int Number of dimensions (this is always 2) format : str Three letter code for the format of the array storage, e.g. 'lil' nnz : int Number of values stored in the array size : int Number of values stored in the array T : lil_array The transpose of the array mT : lil_array The matrix transpose of the array Notes ----- Sparse arrays can be used in arithmetic operations: they support addition, subtraction, multiplication, division, and matrix power. Advantages of the LIL format - supports flexible slicing - changes to the array sparsity structure are efficient Disadvantages of the LIL format - arithmetic operations LIL + LIL are slow (consider CSR or CSC) - slow column slicing (consider CSC) - slow matrix vector products (consider CSR or CSC) Intended Usage - LIL is a convenient format for constructing sparse arrays - once an array has been constructed, convert to CSR or CSC format for fast arithmetic and matrix vector operations - consider using the COO format when constructing large arrays Data Structure - An array (``self.rows``) of rows, each of which is a sorted list of column indices of non-zero elements. - The corresponding nonzero values are stored in similar fashion in ``self.data``. Nrrrrr@rEr=rrsDrErceZdZdZy)ra Row-based LIst of Lists sparse matrix. This is a structure for constructing sparse matrices incrementally. Note that inserting a single item can take linear time in the worst case; to construct the matrix efficiently, make sure the items are pre-sorted by index, per row. .. warning:: SciPy sparse is shifting from a sparse matrix interface to a sparse array interface. In the next few releases we expect to deprecate the sparse matrix interface. For documentation of the matrix interface, see the :ref:`spmatrix interface docs `. For guidance on converting existing code to sparse arrays, see :ref:`Migration from spmatrix to sparray `. This can be instantiated in several ways: lil_matrix(D) where D is a 2-D ndarray lil_matrix(S) with another sparse array or matrix S (equivalent to S.tolil()) lil_matrix((M, N), [dtype]) to construct an empty matrix with shape (M, N) dtype is optional, defaulting to dtype='d'. Attributes ---------- data : ndarray LIL format data array of the matrix rows : ndarray LIL format row index array of the matrix dtype : dtype Data type of the matrix shape : 2-tuple Shape of the matrix ndim : int Number of dimensions (this is always 2) format : str Three letter code for the format of the matrix storage, e.g. 'lil' nnz : int Number of values stored in the matrix size : int Number of values stored in the matrix T : lil_matrix The transpose of the matrix mT : lil_array The transpose of the matrix Notes ----- Sparse matrices can be used in arithmetic operations: they support addition, subtraction, multiplication, division, and matrix power. Advantages of the LIL format - supports flexible slicing - changes to the matrix sparsity structure are efficient Disadvantages of the LIL format - arithmetic operations LIL + LIL are slow (consider CSR or CSC) - slow column slicing (consider CSC) - slow matrix vector products (consider CSR or CSC) Intended Usage - LIL is a convenient format for constructing sparse matrices - once a matrix has been constructed, convert to CSR or CSC format for fast arithmetic and matrix vector operations - consider using the COO format when constructing large matrices Data Structure - An array (``self.rows``) of rows, each of which is a sorted list of column indices of non-zero elements. - The corresponding nonzero values are stored in similar fashion in ``self.data``. Nrr@rEr=rrVsMrErr?)r __docformat____all__bisectrnumpyr-_matrixr_baser r r _indexr r r_sputilsrrrrrrrrrrrr@rEr=rsv& 7--<<$$j*j*Z &R %HE 7EPN9NrE