3 5^* @sdZddlmZmZmZddlZddlmZye e ej fZ Wne k rZe ej fZ YnXddZGdd d eZd d Zd d ZddZdS)z*Indexing mixin for sparse matrix classes. )divisionprint_functionabsolute_importN) isintlikecCs0tj||\}}|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 2lib/python3.6/site-packages/scipy/sparse/_index.py_broadcast_arrayss   rc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(S)) IndexMixinzS This class provides common dispatching and validation logic for indexing. cCs|j|\}}t|trft|tr.|j||St|trD|j||S|jdkrZ|j||Stdnt|trt|tr|j ||St|tr|tdkr||kr|j S|j ||S|jdkr|j ||Stdn|jdkrt|tr|j ||St|trx|j||Snbt|tr.|j ||St|trDtdn4|jddkrx|jdkrx|j|dddf|St||\}}|j|jkrtd|jdkr|jtj|j|jdS|j||S)Nrzindex results in >2 dimensionsrz'number of row and column indices differ)dtype)_validate_indices isinstance INT_TYPES _get_intXintslice_get_intXslicendim_get_intXarray IndexError_get_sliceXintcopy_get_sliceXslice_get_sliceXarray_get_arrayXint_get_arrayXsliceshape_get_columnXarrayrsize __class__rZ atleast_2dr_get_arrayXarray)selfkeyrowcolr r r __getitem__"sH                         zIndexMixin.__getitem__c Cs8|j|\}}t|tr\t|tr\tj||jd}|jdkrDtd|j|||j ddSt|t rtj |j |j ddddf}n tj|}t|t rtj |j |j ddddf}|jdkr|dddf}n tj|}t||\}}|j |j kr tdddlm}||r|jdkr<|d}|d}|j ddkoX|j ddk}|j ddkov|j ddk} |s|j d|j dko| p|j d|j dkstd|jdkrdS|jdd }|j|j|||nHtj||jd}t||\}} |jdkrdS|j|j }|j|||dS) N)rrz&Trying to assign a sequence to an itemrz'number of row and column indices differ) isspmatrixzshape mismatch in assignmentT)r)rrrrasarrayrr# ValueError _set_intXintZflatrZarangeindicesr!Z atleast_1drrrbaser+ZtocooZsum_duplicates_set_arrayXarray_sparseZreshape_set_arrayXarray) r&r'r r(r)ijr+Z broadcast_rowZ broadcast_col_r r r __setitem__MsN  $  "         zIndexMixin.__setitem__cCs|j\}}t|\}}t|rVt|}|| ks8||krDtd||dkrl||7}nt|tsl|j||}t|rt|}|| ks||krtd||dkr||7}nt|ts|j||}||fS)Nzrow index (%d) out of rangerzcolumn index (%d) out of range)r! _unpack_indexrintrrr _asindices)r&r'MNr(r)r r rrs&          zIndexMixin._validate_indicesc Csytj|}Wn"tttfk r0tdYnX|jdkrDtd|jdkrR|S|j}||krntd||j }|dkr|| krtd|||ks|j j r|j }||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 indexrzIndex dimension must be <= 2rzindex (%d) out of range)rr<) rr,r- TypeError MemoryErrorrrr#maxminrZowndatar)r&idxZlengthr Zmax_indxZmin_indxr r rr9s&     zIndexMixin._asindicescCsP|j\}}t|}|| ks$||kr0td||dkr@||7}|j|tdS)zGReturn a copy of row i of the matrix, as a (1 x n) row vector. zindex (%d) out of rangerN)r!r8rrr)r&r3r:r;r r rgetrows  zIndexMixin.getrowcCsP|j\}}t|}|| ks$||kr0td||dkr@||7}|jtd|S)zMReturn a copy of column i of the matrix, as a (m x 1) column vector. zindex (%d) out of rangerN)r!r8rrr)r&r3r:r;r r rgetcols  zIndexMixin.getcolcCs tdS)N)NotImplementedError)r&r(r)r r rrszIndexMixin._get_intXintcCs tdS)N)rD)r&r(r)r r rrszIndexMixin._get_intXarraycCs tdS)N)rD)r&r(r)r r rrszIndexMixin._get_intXslicecCs tdS)N)rD)r&r(r)r r rrszIndexMixin._get_sliceXintcCs tdS)N)rD)r&r(r)r r rrszIndexMixin._get_sliceXslicecCs tdS)N)rD)r&r(r)r r rrszIndexMixin._get_sliceXarraycCs tdS)N)rD)r&r(r)r r rrszIndexMixin._get_arrayXintcCs tdS)N)rD)r&r(r)r r rr szIndexMixin._get_arrayXslicecCs tdS)N)rD)r&r(r)r r rr"szIndexMixin._get_columnXarraycCs tdS)N)rD)r&r(r)r r rr%szIndexMixin._get_arrayXarraycCs tdS)N)rD)r&r(r)r r r rr.szIndexMixin._set_intXintcCs tdS)N)rD)r&r(r)r r r rr2szIndexMixin._set_arrayXarraycCs4tj|j|jd}t||\}}|j|||dS)N)r)rr,Ztoarrayrrr2)r&r(r)r r5r r rr1sz"IndexMixin._set_arrayXarray_sparseN)__name__ __module__ __qualname____doc__r*r6rr9rBrCrrrrrrrr r"r%r.r2r1r r r rrs(+3  rcCsddlm}m}t||tjfr>|jdkr>|jjdkr>|j St |}t|t rt |dkrf|\}}qt |dkr|dt d}}qtdn|t d}}||s||rtdt|tjr|jjdkrt|}t|tjr|jjdkrt|}||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+r<r rNzinvalid number of indiceszoIndexing with sparse matrices is not supported except boolean indexing where matrix and index are equal shapes.)r0rIr+rrZndarrayrrZkindnonzero_check_ellipsistuplelenrr_boolean_index_to_array)indexrIr+r(r)r r rr7s(     r7cCs&|tkrtdtdfSt|ts&|Sx&t|D]\}}|tkr0|}Pq0W|St|dkrjtdtdfSt|dkr|dkr|dtkrtdtdfStd|dfS|dtdfSg}x*||ddD]}|tk r|j|qW|t|}tdd|}|d|tdf|t|S)z6Process indices with Ellipsis. Returns modified index.Nrr<r)EllipsisrrrL enumeraterMappendr?)rOr4vZfirst_ellipsistailZndZnslicer r rrKs0     rKcCs|jdkrtd|jdS)Nrzinvalid index shaper)rrrJ)rAr r rrNEs rN)rHZ __future__rrrZnumpyrZsputilsrr8ZlongZintegerr NameErrorrobjectrr7rKrNr r r rs Z''