a Re2@szdZddlZddlmZddlmZeejfZ ddZ Gdd d Z d d Z d d Z ddZdddZddZddZdS)z*Indexing mixin for sparse matrix classes. N)warn) isintlikecCs0t||\}}|jj|j_|jj|j_||fS)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. )npZbroadcast_arraysflagsZ writeable)abxyr G/var/www/sunrise/env/lib/python3.9/site-packages/scipy/sparse/_index.py_broadcast_arrays s   r c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*S)+ IndexMixinzS This class provides common dispatching and validation logic for indexing. cCs|jrtddS)zWe do not currently support 1D sparse arrays. This function is called each time that a 1D array would result, raising an error instead. Once 1D sparse arrays are implemented, it should be removed. zcWe have not yet implemented 1D sparse slices; please index using explicit indices, e.g. `x[:, [0]]`N)Z _is_arrayNotImplementedError)selfr r r _raise_on_1d_array_slicesz#IndexMixin._raise_on_1d_array_slicecCs||\}}t|trt|tr.|||St|trL||||S|jdkrj||||S|jdkr|||St dn:t|tr t|tr|| ||St|tr|tdkr||kr| S| ||S|jdkr| ||St dn|jdkrPt|tr6||||St|tr|||Snvt|trh|||St|tr~t dnH|jddkr|jdks|jddkr||dddf|St||\}}|j|jkrt d|jdkr|jt|j|jdS|||S)Nrzindex results in >2 dimensionsr'number of row and column indices differdtype)_validate_indices isinstance INT_TYPES _get_intXintslicer_get_intXslicendim_get_intXarray IndexError_get_sliceXintcopy_get_sliceXslice_get_sliceXarray_get_arrayXint_get_arrayXsliceshape_get_columnXarrayZravelr size __class__rZ atleast_2dr_get_arrayXarray)rkeyrowcolr r r __getitem__+sT                          , zIndexMixin.__getitem__c Csb||\}}t|tr\t|tr\tj||jd}|jdkrDtd||||j ddSt|t rtj | |j ddddf}n t|}t|t rtj | |j ddddf}|jdkr|dddf}n t|}t||\}}|j |j kr tdddlm}||r|jdkr<|d}|d}|j ddkoX|j ddk}|j ddkov|j ddk} |s|j d|j dkr| s|j d|j dkstd|j ddks|j ddkrdS|jdd }|||||n^tj||jd}|j |j kr4t||j }|jdkrDdS||j }||||dS) Nrrz&Trying to assign a sequence to an itemrr)issparsezshape mismatch in assignmentT)r )rrrrasarrayrr' ValueError _set_intXintZflatrZarangeindicesr%Z atleast_1drr r_baser.ZtocooZsum_duplicates_set_arrayXarray_sparseZsqueezeZ broadcast_toZreshape_set_arrayXarray) rr*r r+r,ijr.Z broadcast_rowZ broadcast_colr r r __setitem__]sV  $  "         zIndexMixin.__setitem__cCs|j\}}t|\}}t|rVt|}|| ks8||krDtd||dkrl||7}nt|tsl|||}t|rt|}|| ks||krtd||dkr||7}nt|ts|||}||fS)Nzrow index (%d) out of rangerzcolumn index (%d) out of range)r% _unpack_indexrintrrr _asindices)rr*MNr+r,r r r rs&          zIndexMixin._validate_indicesc Cszt|}Wn4tttfyB}ztd|WYd}~n d}~00|jdvrVtd|jdkrd|S|}||krtd|| }|dkr|| krtd|||us|j j s| }||dk|7<|S)zConvert `idx` to a valid index for an axis with a given length. Subclasses that need special validation can override this method. z invalid indexN)rrzIndex dimension must be 1 or 2rindex (%d) out of range) rr/r0 TypeError MemoryErrorrrr'maxminrZowndatar )ridxlengthr eZmax_indxZmin_indxr r r r;s&      zIndexMixin._asindicescCsP|j\}}t|}|| ks$||kr0td||dkr@||7}||tdS)zGReturn a copy of row i of the matrix, as a (1 x n) row vector. r>rN)r%r:rrrrr6r<r=r r r _getrows  zIndexMixin._getrowcCsP|j\}}t|}|| ks$||kr0td||dkr@||7}|td|S)zMReturn a copy of column i of the matrix, as a (m x 1) column vector. r>rN)r%r:rrrrFr r r _getcols  zIndexMixin._getcolcCs tdSNrrr+r,r r r rszIndexMixin._get_intXintcCs tdSrIrJrKr r r rszIndexMixin._get_intXarraycCs tdSrIrJrKr r r rszIndexMixin._get_intXslicecCs tdSrIrJrKr r r rszIndexMixin._get_sliceXintcCs tdSrIrJrKr r r r!szIndexMixin._get_sliceXslicecCs tdSrIrJrKr r r r"szIndexMixin._get_sliceXarraycCs tdSrIrJrKr r r r#szIndexMixin._get_arrayXintcCs tdSrIrJrKr r r r$szIndexMixin._get_arrayXslicecCs tdSrIrJrKr r r r&szIndexMixin._get_columnXarraycCs tdSrIrJrKr r r r)szIndexMixin._get_arrayXarraycCs tdSrIrJrr+r,r r r r r1szIndexMixin._set_intXintcCs tdSrIrJrLr r r r5szIndexMixin._set_arrayXarraycCs4tj||jd}t||\}}||||dS)Nr)rr/Ztoarrayrr r5)rr+r,r _r r r r4sz"IndexMixin._set_arrayXarray_sparseN)__name__ __module__ __qualname____doc__rr-r8rr;rGrHrrrrr!r"r#r$r&r)r1r5r4r r r r rs*24  rcCs0ddlm}m}t||tjfr>|jdkr>|jjdkr>| St |}t|t rt |dkrf|\}}qt |dkr|dt d}}qtdnLt|}|dur|t d}}n,|jdkrt|t dfS|jdkr| S||s||rtdt|}t|}|durt|}|dur(t|}||fS) z Parse index. Always return a tuple of the form (row, col). Valid type for row/col is integer, slice, or array of integers. r)_spbaser.rrrNzinvalid number of indiceszoIndexing with sparse matrices is not supported except boolean indexing where matrix and index are equal shapes.)r3rRr.rrZndarrayrrkindZnonzero_check_ellipsistuplelenrr_compatible_boolean_index_boolean_index_to_array)indexrRr.r+r,rCZbool_rowZbool_colr r r r9s@          r9cCs6|turtdtdfSt|ts&|Sddt|D}|s@|St|dkrZtdtdd|d}t|dkr~tdtdfSt|dkr|dkr|dturtdtdfStd|dfS|dtdfSg}||ddD]}|tur||q|t|}t dd|}|d|tdf|t|S) z6Process indices with Ellipsis. Returns modified index.NcSsg|]\}}|tur|qSr )Ellipsis).0r6vr r r ?z#_check_ellipsis..rz?multi-Ellipsis indexing is deprecated will be removed in v1.13.r) stacklevelr) rZrrrU enumeraterVrDeprecationWarningappendrA)rYZellipsis_indicesZfirst_ellipsistailr\ZndZnslicer r r rT6s6       rTcCst|}|jjdkr|SdS)z8Returns a compatible array if elements are boolean. rN)rZ asanyarrayrrSrCr r r _maybe_bool_ndarray[s  rercCsP|dkr dSztt|d}Wnty2YdS0t|trBdSt||dS)zQReturns True if first element of the incompatible array type is boolean. rNT)nextiterr?rbool_first_element_bool)rCZmax_dimfirstr r r rids  ricCst|dst|rt|SdS)ztReturns a boolean index array that can be converted to integer array. Returns None if no such array exists. rN)hasattrrirerdr r r rWssrWcCs |jdkrtdt|dS)Nrzinvalid index shaper)rrrwhererdr r r rX}s rX)r)rQnumpyrwarningsrZ_sputilsrr:integerrr rr9rTrerirWrXr r r r s   p.%