o i?@sTdZddlZddlmZddlmZmZeej fZ ddZ Gdd d Z d d Z dS) z0Indexing mixin for sparse array/matrix classes. N) isintlike)sparrayissparsecCs0t||\}}|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 M/home/kim/smarthome/.venv/lib/python3.10/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*d+Zd,d-Zd.d/Zd0d1Zd2S)3 IndexMixinzS This class provides common dispatching and validation logic for indexing. cCst||\}}t|dkrq|d}t|tjr |jdkr |}t|tr+||}nt|t r6| |}n| |}t|t sB|S|jdkrl|dkrlt|dkr[|j |g||jdSt|dkrl|j |gg||jdS||S|\}}t|trt|tr|||}nt|t r|||}n|jdkr|||}n|jdkr|||}ntdt|t rt|tr|||}nt|t r|t dkr||kr|}n|||}n|jdkr|||}nstdt|tr|||}nct|t r|||}nV|jdkr.|jddkr.|jdks|jddkr.||dddf|}n+t||\}}|j|jkr@td|jdkrS|j t|j|jd}n| ||}t|t s|dksqt|dkrs|jdkrs|St|dkr~d |}||jkr|S||S|j|kr|j!d krt|dkr|jdkr|j"|g|d S|#|S||S|S) Nrrr )shapedtypezindex results in >2 dimensions'number of row and column indices differr)rZlilr)$_validate_indiceslen isinstancerZndarrayritem 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_columnXarrayZravelrsizeZ atleast_2d_get_arrayXarrayformatZ_coo_containertocoo)selfkeyindexZ new_shapeidxresrowcolr r r __getitem__s                           $   zIndexMixin.__getitem__cCsp||\}}t|dkr|d}t|r|}ntj||jd}t|tr;|j dkr0t d| ||j ddSt|t rst||jd}t|}|dkrd|j dkrd| |d|j ddStj||jd}|j}n|j}|j|krt||j}|j dkr|||dS|\} } t| trt| trtj||jd}|j dkrt d|| | |j ddSt| t rtj| |jddddf} nt| } t| t rtj| |jddddf} | jdkr| dddf} nt| } t| | \} } | j| jkrtdt|rd|jvr"dS| jdkr0| d} | d} |jddj|jdd}|jddkoJ| jddk} |jddkoY| jddk}| sh|jd| jdkrv|sz|jd| jdkszt d ||| | |dStj||jd}|j| jkrt|| j}|j dkrdS|| j}|| | |dS) Nrrrz&Trying to assign a sequence to an itemrF)r'Tzshape mismatch in assignment) rrrtoarrayrasarrayrrrr- ValueError_set_intZflatrrangeindicesrZarangeZsqueezeZ broadcast_to _set_array _set_intXintZ atleast_1dr#rr%r0r Z _shape_as_2dZsum_duplicates_set_arrayXarray_sparse_set_arrayXarray)r1r2r r3_r4Z idx_rangeN idx_shaper6r7ijZ broadcast_rowZ broadcast_colr r r __setitem__|s          $  "       zIndexMixin.__setitem__cs|turtdf|j|jfSt|ts|g}d}g}d}t|D]Y\}}|tur4|dur1td|}q!|dur>||q!t|tsGt |rQ|||d7}q!t ||j}durf||||j7}q!t |rntd|t ||d7}q!|j|tdg}|r|dur||n|d||||d}g} d} gg} t|D]\}}|dur| dqt|trۈ||j| } tt|| } | | | d7} qt |r|j| }| |kr|ksntd|dt|dkr||n|}|| d7} q|jjdkrK|}| |j}|j| |}|j|kr9td |d |d |j|| t| ||} q|j| }|||}|| | | d7} q| |jkrwtd |jd | dt| dkrtfdd| DtfddddDrdddD}d|}t|tdj| } n t| dkr| d}t|j}| d||| |d} t| }dkrtd|dtt| fS)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.zindex (z) out of ranger z bool index z has shape z instead of zinvalid index ndim. Array is zD. Index needs Dc3s|]}|VqdSNr ).0rF)r3r r )sz/IndexMixin._validate_indices..c3s |] }dj|jkVqdS)rNrrKix) idx_arraysr r rL*s css|]}t|jVqdSrJ)strrrMr r r rL+szLshape mismatch: indexing arrays could not be broadcast together with shapes rz*Only 1D or 2D arrays allowed. Index makes )Ellipsisrr#rrtuple enumerater%appendr_compatible_boolean_indexrrr:extend_shaperr=r>intrkindZnonzero _asindicesranyjoinlist)r1r2Z ellps_posZ index_1stZ prelim_ndimrFr4rNZ ellip_slicesrEZ index_ndimZ array_indicesZMsZ len_slicerDZtmp_ndimZ mid_shapeZshapesmsgZ arr_indexZ arr_shaper#r )rOr3r rs                           zIndexMixin._validate_indicesc Cszt|}Wntttfy}ztd|d}~ww|jdvr%td|jdkr,|S|}||kr:td|| }|dkrc|| krMtd|||usU|j j sY| }||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 MemoryErrorr%r#r-maxminrZowndatar')r1r4lengthr eZmax_indxZmin_indxr r r r[9s*      zIndexMixin._asindicescCsP|j\}}t|}|| ks||krtd||dkr ||7}||tdS)zGReturn a copy of row i of the matrix, as a (1 x n) row vector. r`rN)rrYr%r"rr1rFMrDr r r _getrowW  zIndexMixin._getrowcCsP|j\}}t|}|| ks||krtd||dkr ||7}|td|S)zMReturn a copy of column i of the matrix, as a (m x 1) column vector. r`rN)rrYr%r&rrgr r r _getcolbrjzIndexMixin._getcolcCtrJNotImplementedErrorr1r4r r r rmzIndexMixin._get_intcCrlrJrmror r r rprpzIndexMixin._get_slicecCrlrJrmror r r rsrpzIndexMixin._get_arraycCrlrJrmr1r6r7r r r r!vrpzIndexMixin._get_intXintcCrlrJrmrqr r r r$yrpzIndexMixin._get_intXarraycCrlrJrmrqr r r r"|rpzIndexMixin._get_intXslicecCrlrJrmrqr r r r&rpzIndexMixin._get_sliceXintcCrlrJrmrqr r r r(rpzIndexMixin._get_sliceXslicecCrlrJrmrqr r r r)rpzIndexMixin._get_sliceXarraycCrlrJrmrqr r r r*rpzIndexMixin._get_arrayXintcCrlrJrmrqr r r r+rpzIndexMixin._get_arrayXslicecCrlrJrmrqr r r r,rpzIndexMixin._get_columnXarraycCrlrJrmrqr r r r.rpzIndexMixin._get_arrayXarraycCrlrJrmr1r4r r r r r<rpzIndexMixin._set_intcCrlrJrmrrr r r r?rpzIndexMixin._set_arraycCrlrJrmr1r6r7r r r r r@rpzIndexMixin._set_intXintcCrlrJrmrsr r r rBrpzIndexMixin._set_arrayXarraycCs4tj||jd}t||\}}||||dS)Nr)rr:r9rrrB)r1r6r7r rCr r r rAsz"IndexMixin._set_arrayXarray_sparseN)__name__ __module__ __qualname____doc__r8rHrr[rirkrrrr!r$r"r&r(r)r*r+r,r.r<r?r@rBrAr r r r rs4_Xe   rcCst|ds6z!tt|d}t|D]}t|trn tt|d}qWdSWn ty0YdSwt|}|j j dkr>|SdS)zICheck for boolean array or array-like. peek before asarray for array-liker#Nr ) hasattrnextiterr=rboolrarZ asanyarrayrrZ)r4Z desired_ndimrNrCr r r rVs      rV)rwnumpyrZ_sputilsrZ_baserrrYintegerrrrrVr r r r s