p:i?hSrSSKrSSKJr SSKJrJr \\R4r Sr "SS5r S r g) z0Indexing mixin for sparse array/matrix classes. N) isintlike)sparrayissparsec[R"X5up#URRURlURRURlX#4$)a Same as np.broadcast_arrays(a, b) but old writeability rules. NumPy >= 1.17.0 transitions broadcast_arrays to return read-only arrays. Set writeability explicitly to avoid warnings. Retain the old writeability rules, as our Cython code assumes the old behavior. )npbroadcast_arraysflags writeable)abxys V/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/scipy/sparse/_index.py_broadcast_arraysr sI   q $DA))AGG))AGG 4Kc\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rSrSrSrSrSrSrSrSrSrSrSrSrSrSrg) IndexMixinzK This class provides common dispatching and validation logic for indexing. c URU5up#[U5S:XGa6USn[U[R5(a UR S:XaUR 5n[U[5(aURU5nO8[U[5(aURU5nOURU5n[U[5(dU$UR S:Xa[US:waU[U5S:XaURU/X0RS9$[U5S:XaURU//X0RS9$URU5$Uupg[U[5(a[U[5(aUR!Xg5nGO{[U[5(aUR#Xg5nGOSUR$S:XaUR'Xg5nGO0UR$S:XaUR'Xg5nGO [)S5e[U[5(a[U[5(aUR+Xg5nGO[U[5(a9U[S5:XaXg:XaUR-5nGOUR/Xg5nGOwUR$S:XaUR1Xg5nGOT[)S5e[U[5(aUR3Xg5nGO![U[5(aUR5Xg5nOUR$S:Xa^UR SS:XaKUR$S:XdUR SS:Xa(UR7USS2S4UR955nO[;Xg5upgUR UR :wa [)S5eUR<S:Xa9UR[R>"U5R URS9nOURAXg5n[U[5(d]US:Xd[U5S:XaUR$S:waU$[U5S:XaS U-nX5R :XaU$URU5$UR U:wapURBS :XaO[U5S:wa@UR S:XaUREU/US 9$URG5RU5$URU5$U$) Nrr)shapedtypezindex results in >2 dimensions'number of row and column indices differr)rlilr)$_validate_indiceslen isinstancerndarrayritem INT_TYPES_get_intslice _get_slice _get_arrayr __class__rreshape _get_intXint_get_intXslicendim_get_intXarray IndexError_get_sliceXintcopy_get_sliceXslice_get_sliceXarray_get_arrayXint_get_arrayXslice_get_columnXarrayravelrsize atleast_2d_get_arrayXarrayformat_coo_containertocoo)selfkeyindex new_shapeidxresrowcols r __getitem__IndexMixin.__getitem__s11#6 u:?(C#rzz**99?((*C#y))mmC(C''ooc*ooc*dG,, yyB9?y>Q&>>3%y >SSy>Q&>>C5'**>UU;;y) ) c9 % %#y))''1C''))#3Q))#3Q))#3 !ABB U # ##y))))#3C''%+%#*))+C//9CQ++C5 !ABB#y))))#3C''++C5((a-CIIaLA$5XX]ciila&7,,SAY D-S699 )$%NOO88q=..s);)A)A.TC//9C$((B3y>Q#6388q= 9~" 9, #yy03 Lckk)6L L 99 !{{e#I!(;99?..ui.HHyy{**955;;y) ) rc URU5up4[U5S:XGaUSn[U5(aUR5nO[R "X R S9n[U[5(a:URS:wa [S5eURXRRS5 g[U[5(a[URUR S56n[U5nUS:Xa3URS:Xa#URUSURS5 g[R""URUR S56nUR nOUR%5R nUR%5R U:wa [R&"X%R 5nURS:waUR)XR5 gUup[U [5(an[U [5(aY[R "X R S9nURS:wa [S5eUR+XURS5 g[U [5(a7[R""U RUR S56SS2S4n O[R,"U 5n [U [5(aP[R""U RUR S56SSS24n U R.S:Xa U SS2S4n O[R,"U 5n [1X5upU R U R :wa [3S5e[U5(Ga$SUR ;agU R.S:Xa U Sn U Sn UR5SS9R7UR8SS9nUR SS:H=(a U R SS:gn UR SS:H=(a U R SS:gnU (d UR SU R S:Xa'U(d+UR SU R S:Xd [S 5eUR;5 UR=XU5 g[R "X R S9nUR%5R U R%5R :wa [R&"X+R 5nURS:XagUR7U R 5nUR?XU5 g) Nrrrz&Trying to assign a sequence to an itemrF)r1Tzshape mismatch in assignment) rr rtoarrayrasarrayrr!r$r8 ValueError_set_intflatr&rangeindicesrarangesqueeze broadcast_to _set_array _set_intXint atleast_1dr-rr/r=r* _shape_as_2dsum_duplicates_set_arrayXarray_sparse_set_arrayXarray)r>r?rr@_rB idx_rangeN idx_shaperDrEij broadcast_row broadcast_cols r __setitem__IndexMixin.__setitem__|s))#. u:?(C{{IIKJJq 3#y))66Q;$%MNN c66!9-#u%%"3;;tzz!}#=>  N6affkMM)A,q :iiTZZ]!;<II KKM// yy{  I-OOAyy1vv{'  c9 % %*S)*D*D 1JJ/Avv{ !IJJ   cq 2  c5 ! !))S[[A78DAC--$C c5 ! !))S[[A78qACxx1}!T'l--$C * 77agg FG G A;;AGG|vv{dGdGU#++ANN+FAGGAJ!O? aMGGAJ!O? aM"aggajAGGAJ&>"aggajAGGAJ&> !?@@     ( (q 1 1JJ/Ayy{  AIIK$5$55OOAww/vv{ !''"A  ! !! *rc P ^^U[La%[S54UR-UR4$[ U[ 5(dU/nSn/nSn[ U5HupVU[LaUb [S5eUnM UcURU5 M6[ U[5(d[U5(aURU5 US- nMs[X`R5=nb!URU5 XGR- nM[U5(a [S5eUR[R"U55 US- nM URU- [S5/-nU(a"UcURU5 O USUU-X2S-n/n Sn /m/n [ U5GHupVUcU RS5 M[ U[5(aXTRU5 URU n [![#UR%U 565n U RU 5 U S- n M[U5(a[URU nU*Us=::aU:dO [SUS35e['US:aXn-OU5nTRU5 U S- n MUR(R*S:XaUnXR-nURXnURU:wa[S US US UR35eTRUR-55 U R[#X55 Un GMURU nUR/Xn5nTRU5 U RU 5 U S- n GM XR:a[S URS U S35e[!U 5S:as[1U4SjU 56m[3U4SjTSS55(a(SR5ST55nSU3n[U5e[7TSR5U -n O:[!U 5S:Xa+U Sn[7TUR5nU SUU-U US-n [!U 5=nS:a[SUS35e[ T5[ U 54$)z8Returns two tuples: (index tuple, requested shape tuple)Nrz(an index can only have a single ellipsisrzoIndexing with sparse matrices is not supported except boolean indexing where matrix and index are equal shapes.index () out of ranger z bool index z has shape z instead of zinvalid index ndim. Array is zD. Index needs Dc3.># UH nTUv M g7fNr).0r^r@s r /IndexMixin._validate_indices..)s,M}!U1X}sc3\># UH!nTSRUR:gv M# g7f)rNr)rjix idx_arrayss rrkrl*s$L^r:a=&&"((2^s), c3L# UHn[UR5v M g7fri)strr)rjrns rrkrl+s!E*B#bhh--*s"$zLshape mismatch: indexing arrays could not be broadcast together with shapes rz*Only 1D or 2D arrays allowed. Index makes )Ellipsisr&r-rr!tuple enumerater/appendr_compatible_boolean_indexrrrJextend_shaper rNrOintrkindnonzero _asindicesranyjoinlist)r>r? ellps_pos index_1st prelim_ndimr^rBrn ellip_slicesr] index_ndim array_indicesMs len_slicer\tmp_ndim mid_shapeshapesmsg arr_index arr_shaper-ror@s @@rrIndexMixin._validate_indicessc (?$K>DII-tzz9 9#u%%%C   nFAh($%OPP   %C''9S>>  %q 1#yyAA"N  $ww& # ())   C1q +%, K/E$K=@    .%jy1L@9ZCXX     *FA{  #C'' S![[,s{{2 78   +a 3KK +c A $wse>%BCCS1W#'#6 S!a 3&%/ KK < 88y($%aS I;l288*U RZZ\*$$U:%@A% KK +ooc- S!$$Z0a C+D  !/ {/*UVW  }  !*,M},MNJLZ^LLL!E*!EE//5h8 o%Z]001I=I  1 $%a(IU9-334I!*9- 9IijrBlengthremax_indxmin_indxs rr}IndexMixin._asindices9s  5 3A 66 => > 66Q;H557  wxj?@ @557 a<'! 78*N!CDDxqwwFFH !eH H+I{3 5_-1 4 5sCC:) C55C:cURup#[U5nX*:dX:a[SUS35eUS:aX- nURU[ S55$)zGReturn a copy of row i of the matrix, as a (1 x n) row vector. rerfrN)rrzr/r,r&r>r^Mr\s r_getrowIndexMixin._getrowWs\zz F r6QVwqc89 9 q5 FA""1eDk22rcURup#[U5nX*:dX:a[SUS35eUS:aX- nUR[ S5U5$)zMReturn a copy of column i of the matrix, as a (m x 1) column vector. rerfrN)rrzr/r0r&rs r_getcolIndexMixin._getcolbs\zz F r6QVwqc89 9 q5 FA""5;22rc[5eriNotImplementedErrorr>rBs rr%IndexMixin._get_intm !##rc[5erirrs rr'IndexMixin._get_sliceprrc[5erirrs rr(IndexMixin._get_arraysrrc[5erirr>rDrEs rr+IndexMixin._get_intXintvrrc[5erirrs rr.IndexMixin._get_intXarrayyrrc[5erirrs rr,IndexMixin._get_intXslice|rrc[5erirrs rr0IndexMixin._get_sliceXintrrc[5erirrs rr2IndexMixin._get_sliceXslicerrc[5erirrs rr3IndexMixin._get_sliceXarrayrrc[5erirrs rr4IndexMixin._get_arrayXintrrc[5erirrs rr5IndexMixin._get_arrayXslicerrc[5erirrs rr6IndexMixin._get_columnXarrayrrc[5erirrs rr:IndexMixin._get_arrayXarrayrrc[5erirr>rBrs rrLIndexMixin._set_intrrc[5erirrs rrSIndexMixin._set_arrayrrc[5erirr>rDrErs rrTIndexMixin._set_intXintrrc[5erirrs rrYIndexMixin._set_arrayXarrayrrc[R"UR5URS9n[ X15up4UR XU5 g)Nr)rrJrIrrrY)r>rDrErrZs rrX"IndexMixin._set_arrayXarray_sparses8 JJqyy{$** 5 ( c*rrN)__name__ __module__ __qualname____firstlineno____doc__rFrbrr}rrr%r'r(r+r.r,r0r2r3r4r5r6r:rLrSrTrYrX__static_attributes__rrrrrs]~V+pc.J< 3 3$$$$$$$$$$$$$$$$$+rrcT[US5(dk[[U5S5n[U5H/n[ U[ 5(a O[[U5S5nM1 g[R"U5nURRS:XaU$g![ a gf=f)zICheck for boolean array or array-like. peek before asarray for array-liker-Nr ) hasattrnextiterrNr!boolrr asanyarrayrr{)rB desired_ndimrnrZs rrwrws 3   d3i&B<(b$''$r(D)) mmC  yy~~   s9B B B'&B')rnumpyr_sputilsr_baserrrzintegerr$rrrwrrrrs8$ "**   K+K+\ r