U ˻|e2@sdZddlZddlmZzeeejfZWne k rFeejfZYnXddZ GdddZ d d Z d d Z d dZdddZddZddZdS)z*Indexing mixin for sparse matrix classes. N) 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. )npbroadcast_arraysflags writeable)abxyr P/var/www/website-v5/atlas_env/lib/python3.8/site-packages/scipy/sparse/_index.py_broadcast_arrays s   rc@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) _is_arrayNotImplementedError)selfr r r _raise_on_1d_array_slice sz#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_columnXarrayravelrsize __class__r 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) isspmatrixzshape mismatch in assignmentT)r")rrrrasarrayrr* ValueError _set_intXintflatrarangeindicesr' atleast_1drrr _baser2tocoosum_duplicates_set_arrayXarray_sparsesqueeze broadcast_toreshape_set_arrayXarray) rr.r r/r0ijr2 broadcast_row 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_indexrintr rr _asindices)rr.MNr/r0r r r rs&          zIndexMixin._validate_indicesc Cszt|}Wn2tttfk r@}ztd|W5d}~XYnX|jdkrTtd|jdkrb|S|}||kr~td|| }|dkr|| krtd|||ks|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) rr3r4 TypeError MemoryErrorr rr*maxminrowndatar")ridxlengthr eZmax_indxZmin_indxr r r rIs&     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. rLrN)r'rHr rrrrBrJrKr 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. rLrN)r'rHr r!rrUr r r getcols  zIndexMixin.getcolcCs tdSNrrr/r0r r r rszIndexMixin._get_intXintcCs tdSrXrYrZr r r rszIndexMixin._get_intXarraycCs tdSrXrYrZr r r rszIndexMixin._get_intXslicecCs tdSrXrYrZr r r r!szIndexMixin._get_sliceXintcCs tdSrXrYrZr r r r#szIndexMixin._get_sliceXslicecCs tdSrXrYrZr r r r$szIndexMixin._get_sliceXarraycCs tdSrXrYrZr r r r%szIndexMixin._get_arrayXintcCs tdSrXrYrZr r r r&szIndexMixin._get_arrayXslicecCs tdSrXrYrZr r r r(szIndexMixin._get_columnXarraycCs tdSrXrYrZr r r r-szIndexMixin._get_arrayXarraycCs tdSrXrYrr/r0r r r r r5szIndexMixin._set_intXintcCs tdSrXrYr[r r r rAszIndexMixin._set_arrayXarraycCs4tj||jd}t||\}}||||dS)Nr)rr3toarrayrrrA)rr/r0r _r r r r=sz"IndexMixin._set_arrayXarray_sparseN)__name__ __module__ __qualname____doc__rr1rFrrIrVrWrrrr!r#r$r%r&r(r-r5rAr=r r r r rs*24  rcCs0ddlm}m}t||tjfr>|jdkr>|jjdkr>| St |}t|t rt |dkrf|\}}qt |dkr|dt d}}qtdnLt|}|dkr|t d}}n,|jdkrt|t dfS|jdkr| S||s||rtdt|}t|}|dk rt|}|dk r(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)spmatrixr2rr rNzinvalid number of indiceszoIndexing with sparse matrices is not supported except boolean indexing where matrix and index are equal shapes.)r:rbr2rrndarrayrrkindnonzero_check_ellipsistuplelenrr _compatible_boolean_index_boolean_index_to_array)indexrbr2r/r0rRZbool_rowZbool_colr r r rG s@          rGcCs |tkrtdtdfSt|ts&|St|D]\}}|tkr.|}qLq.|St|dkrhtdtdfSt|dkr|dkr|dtkrtdtdfStd|dfS|dtdfSg}||ddD]}|tk r||q|t|}tdd|}|d|tdf|t|S)z6Process indices with Ellipsis. Returns modified index.Nrrr)Ellipsisrrrg enumeraterhappendrO)rkrCvZfirst_ellipsistailndZnslicer r r rf9s0      rfcCst|}|jjdkr|SdS)z8Returns a compatible array if elements are boolean. r N)r asanyarrayrrdrRr r r _maybe_bool_ndarray`s  rtrcCsR|dkr dSztt|d}Wntk r4YdSXt|trDdSt||dS)zQReturns True if first element of the incompatible array type is boolean. rNT)nextiterrMrbool_first_element_bool)rRZmax_dimfirstr r r rxis rxcCst|dst|rt|SdS)ztReturns a boolean index array that can be converted to integer array. Returns None if no such array exists. rN)hasattrrxrtrsr r r rixsricCs |jdkrtdt|dS)Nrzinvalid index shaper)rr rwherersr r r rjs rj)r)ranumpyr_sputilsrrHlongintegerr NameErrorrrrGrfrtrxrirjr r r r s p.'