U Gv f2@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. )npZbroadcast_arraysflagsZ writeable)abxyr 7/tmp/pip-unpacked-wheel-96ln3f52/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_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_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) isspmatrixzshape mismatch in assignmentT)r)rrrrasarrayrr& ValueError _set_intXintZflatrZarangeindicesr$Z atleast_1drr r_baser-ZtocooZsum_duplicates_set_arrayXarray_sparseZsqueezeZ broadcast_toZreshape_set_arrayXarray) rr)rr*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|}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) rr.r/ TypeError MemoryErrorrrr&maxminrZowndatar)ridxlengthreZmax_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$r9rrrrr5r;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$r9rrrrEr r r getcols  zIndexMixin.getcolcCs tdSNrrr*r+r r r rszIndexMixin._get_intXintcCs tdSrHrIrJr r r rszIndexMixin._get_intXarraycCs tdSrHrIrJr r r rszIndexMixin._get_intXslicecCs tdSrHrIrJr r r rszIndexMixin._get_sliceXintcCs tdSrHrIrJr r r r szIndexMixin._get_sliceXslicecCs tdSrHrIrJr r r r!szIndexMixin._get_sliceXarraycCs tdSrHrIrJr r r r"szIndexMixin._get_arrayXintcCs tdSrHrIrJr r r r#szIndexMixin._get_arrayXslicecCs tdSrHrIrJr r r r%szIndexMixin._get_columnXarraycCs tdSrHrIrJr r r r(szIndexMixin._get_arrayXarraycCs tdSrHrIrr*r+rr r r r0szIndexMixin._set_intXintcCs tdSrHrIrKr r r r4szIndexMixin._set_arrayXarraycCs4tj||jd}t||\}}||||dS)Nr)rr.Ztoarrayrr r4)rr*r+r_r r r r3sz"IndexMixin._set_arrayXarray_sparseN)__name__ __module__ __qualname____doc__rr,r7rr:rFrGrrrrr r!r"r#r%r(r0r4r3r r r r r s*24  r cCs0ddlm}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)spmatrixr-rrrNzinvalid number of indiceszoIndexing with sparse matrices is not supported except boolean indexing where matrix and index are equal shapes.)r2rQr-rrZndarrayrrkindZnonzero_check_ellipsistuplelenrr_compatible_boolean_index_boolean_index_to_array)indexrQr-r*r+rBZbool_rowZbool_colr r r r8 s@          r8cCs |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)EllipsisrrrT enumeraterUappendr@)rXr6vZfirst_ellipsistailZndZnslicer r r rS9s0      rScCst|}|jjdkr|SdS)z8Returns a compatible array if elements are boolean. rN)rZ asanyarrayrrRrBr r r _maybe_bool_ndarray`s  r_rcCsR|dkr dSztt|d}Wntk r4YdSXt|trDdSt||dS)zQReturns True if first element of the incompatible array type is boolean. rNT)nextiterr>rbool_first_element_bool)rBZmax_dimfirstr r r rcis rccCst|dst|rt|SdS)ztReturns a boolean index array that can be converted to integer array. Returns None if no such array exists. rN)hasattrrcr_r^r r r rVxsrVcCs |jdkrtdt|dS)Nrzinvalid index shaper)rrrwherer^r r r rWs rW)r)rPZnumpyrZ_sputilsrr9longintegerr NameErrorr r r8rSr_rcrVrWr r r r s p.'