i[\dZdZgdZddlZddlmZmZddlm Z dd l m Z m Z m Z mZdd lmZdd lmZmZmZmZmZmZmZmZdd lmZmZmZmZGd deZdZ dZ!GddeeZ"Gdde eZ#y)zSparse DIAgonal formatzrestructuredtext en) dia_array dia_matrixisspmatrix_diaN) _prune_arraycopy_if_needed)spmatrix)issparse_formats_spbasesparray) _data_matrix)isdense isscalarlikeisshape upcast_chargetdtype get_sum_dtype validateaxis check_shape) dia_matmat dia_matvec dia_matvecs dia_tocsrcheZdZdZddddZdZdZddZejje_ ddZ ejje _ dd Z ejje _ dd Z fd Z d Zfd ZdZdZfdZddZddZej(je_ ddZej*je_ ddZej,je_ ddZej.je_ ddZdZej2je_ xZS) _dia_basediaNmaxprintctj|||t|r|jdk(rP|r|j }|j |_|j |_t|j|_ nl|j|jk(r|r|j }n|j}|j |_|j |_t|j|_ nt|tr5t|r}t||_ tjdt!|t"|_|j%t'|j}tjd||_nY |\}} | t)d|st*}tj,tj.|d|| |_tj.|d |j%t'|| } tj0| |_t||_ n tj4|}t|t6r(|j8dk7rt)d|j8d|j;|||j}|j |_|j |_t|j|_ |-t!|} |j j=| d|_|j j8d k7r t)d|j j8dk7r t)d|j jdt?|j k7r{{e#99;D II #|| )$**5 ;;$++-$ A AFF  yy )!''2 e $t}*$/ HHUHUE,JK  11TZZ1I !xx9= 5$(MD' }()DEE- " bhhtAweRV.W XDI hhtAw-1-B-B#e*-B-U,02G$&==#9DL"-e"4DK Mzz$'$(TYY!^ #DI <<   !=> > 99>>Q :; ; 99??1 T\\!2 2' (:';<"4<<014  ryy& '3t||+< <EF F =Y!5KG$W-145$ M !9$(KK=0C"EFKLM Ms0PP: P7$P22P7: Q"QQ"c t|j\}}t|trdnd}|jj d}d|d|d|j d|jd|d |j d S) Nr:matrixr)r r,r1rr-r(r&nnz)rC_fmt sparse_clsds rJ__repr__z_dia_base.__repr__ds|$++&3 *4 9Wx IIOOA uHZL DJJ<@hhZ1!4J4::,VW Y c|j\}}tj|jjd}||jdddfz }|dk\}|||kz}|||kz}|S)z~Returns a mask of the same shape as self.data, where mask[i,j] is True when data[i,j] corresponds to a stored element.r Nr)r(r3aranger-r.)rCnum_rowsnum_cols offset_indsrowmasks rJ _data_maskz_dia_base._data_maskmso"ZZ(ii  23 DLL400q x  x'( rUc| td|j}tj|j|S)Nz>$,, 6??S499uzz#9; ;#'99uzz#9; ;jju}}= ??; =OOK? #**""1% W [!1S5F!Fyyx!89HA&%**4&A&%**4& w 3{#3s5==7I#I!JJ}Y'?@@ ::mI&>? Xq[ !TYY . !DJJqMKO3TZZ]CAxx[!1%nnTYY ;H Xww& '499QU+; ; 'HWH,-ArrE1BB-HWH,-ArrE1BB-""Hk#:$**"MMrUcht|tst| |S|j |dS)NT)r)r1rsuper _sub_sparser{)rCr __class__s rJrz_dia_base._sub_sparses3%+7&u- -400rUc>|j|j|zSN)r}r-)rCrs rJ _mul_scalarz_dia_base._mul_scalarstyy5011rUcPt|r|j|St|r|jdkDr|j |zSd|j vsd|j vsd|j vrt ||Stj|}|j \}}|j \}}t|jj d|}|jddd|fjtj|j|}|dk(r ||dd|fz}n||k7r tdtj|}||kDr||j dddfz |z} n||j dddf|zz } |dk(rd}n||k7r td||| |fz}|j#|St%|t&r|j |j k7rt ||Stj(|j |j dd\} } } t|jj d|jj d}|j| d|f|j| d|fz}|j+|| f|j S)Nrrr zinconsistent shapesT) assume_uniquereturn_indicesrz)rrrr>toarrayr(rmultiplyr3r9rcr-r@rr8rWr.r}r1r intersect1dr)rCr other_rows other_colsrowscolsrjr-jir.rrrs rJrz_dia_base.multiplysk  ##E* * 5>zzA~||~--DJJ!tzz/Q%++5Ew'..MM%(E%*[[ "J JD$DIIOOA&-A99QU#**2>>$))U+KLDQa!e $t# !677IIaLt8T\\!T'22d:ADLLD1D88A?A4'$%:;;ad #??4( (%+u{{djj/H7#E* * NN4<<)-d D %9  "EJJ$4$4Q$7 8yy2A2&IrrM)BB""D'?$**"EErUc |}tj|jdt|jj |jj }|j jd}|j\}}t||t|j||j|j |j|j|S)Nrr%r ) r3r4r(rr&charr-rrAr.ravel)rCrrtyrjrhris rJ_matmul_vectorz_dia_base._matmul_vectors  HHTZZ]+djjoo78ww||+E F IIOOA jj!1QDLL)1dllDII779aggi )rUcTtj|jd|jdftj|j|}t g|j|jj|j |j|jd|||S)Nrr r%)r3r4r(rr-rr.)rCrrus rJ_matmul_multivectorz_dia_base._matmul_multivector)shh 1 u{{1~6^^DIIu=? 0TZZ 0$))// 04<< 0 0KKN 0$) 0+. 0 rUclt|tst| |Sd|jvsd|jvr-|j |jd|jdfSt g|j|jj|j|j|jd|jj|j|j\}}|j |jt|d|f|jd|jdfS)Nrr ry) r1rr_matmul_sparser(rrr-r.reshaperA)rCrr.r-rs rJrz_dia_base._matmul_sparse0s%+7)%0 0  ?a5;;.&& 1 u{{1~'FG G">DJJ>>#'<<>15>#(;;q>>49JJ4D4D>$)==>38**> ""DLLWr$BG#L$(JJqM5;;q>#BD DrUc|j\}}|jdk(rtj}n t |}|dkrt ||z||}d}|}nt |||z |}|}||z}|jdk7r|d|}|j j\} } ||jvro|| kDrItj| |f|j j} |j | ddd| f<| |_||j |j|k(||f<ytj|j|jjj||_t|| } tj| dz| f|j j} |j | ddd| f<|| d||f<| |_y)Nrr%r ry) r(r>r3infrArcr-r.r4r&appendtyper7) rCvalueskrhrivalues_nn min_index max_index data_rows data_colsr-ms rJ_setdiagz_dia_base._setdiag@szz1 ;;! vvH6{H q5AE1h'AIIAq1uh'AIAI ;;! BQZF#yy 9  9$xxI 6diiooN&*iiQ  ]#  @FDIIdlla'9)<< =99T\\4<<3E3E3J3J13MNDLIy)A88Y]A.diiooFD$(IID"jyj !,2DYy(( )DIrUc*|r|jS|Srr))rCr's rJr0z_dia_base.todiaes 99; KrUc||dk7r td|j\}}t|j}|j }t j t |tjdddf}t j |tj||zdddfz }td||jjdz } t j|jt j|jjd| f|jjf} | ||f} |j| |f||f|S)N)r rzvSparse arrays/matrices do not support an 'axes' parameter because swapping dimensions is the only logical permutation.r%rr )r(r') r8r(r7r.r3rWrAintcr-hstackr4r&r) rCaxesr'rXrYmax_dimr.rc pad_amountr-s rJ transposez_dia_base.transposems3  LM M"ZZ(djj/<<- IIc'l"'' 21d7 ; IIhbgg .'G2CQW1M MGDIIOOA$667 yy$))RXXtyyq/A:.N48IIOO&EFGAqDz""D'? h; &*#, ,rUc|j\}}|| ks||k\r+tjd|jjStj |j |k(\}td|}t||z|}||z }|jdk(r+tj||jjS|j|d||f}|t|z } | dkDrtj|d| fd}|S)Nrr%constant)mode) r(r3emptyr-r&nonzeror.r7rcsizer4rApad) rCrrridx first_collast_col result_sizeresultpaddings rJdiagonalz_dia_base.diagonalsZZ d :d88ATYY__5 5zz$,,!+,1I tax&* 88q=88Ktyy? ?3q69X#556F + Q;VVFQLzBF rUcd|jvst|jdk(r'|j|j|jS|j\}}|j }|j t|||}tj|jj|d}tj||j}tj||}tjd|z|} t||g|jj|jj|d|j|||| } |j t| ||}t|d| }t|d| j|d}| j|d} |j||| f|j|j} d| _| S) Nrr%r#Fr)r )r(r&T)r(rAr._csr_containerr&rOr6r7r3argsortr@rrr-rhas_canonical_format) rCr'n_rowsn_colsmax_nnzrFordercsr_dataindicesindptrrOrns rJtocsrz_dia_base.tocsrs  ?c$,,/14&&tzz&D D(( ))Wff1M)N  4<<(// /F88G4::6((7)4!f*I6:: ++IE+B:DHII::':)0:28:))S&&1I)J #/wt}33IE3JKyu5!!8Wf"=(, $**"F#'  rUc|r7|j||jjf|jS|j||jf|jS)zReturns a matrix with the same sparsity structure as self, but with different data. By default the structure arrays are copied. rz)rr.r'r()rCr-r's rJr}z_dia_base._with_datasf &&t||((*+4::' &&t||$DJJ' rUct|}|\}}|jddd|f|_||jdkDrtj|j |jdz|jjdkr_|j dddf|jdztj |jjdk}d|j|<||_y)Nrr )rr-r(r3anyr.rWr/)rCr(rhrir\s rJresizez_dia_base.resizesE"1IIa!e$  1 t||djjm3diiooa6HHILLD)DJJqM9IIdiiooa012DDIIdO rU)NNFr)NNN)Frm)NF)T)__name__ __module__ __qualname___formatr+rTr]r`r __doc__rkrgr{rrrrrrrr0rrrr}r __classcell__)rs@rJrrs"GKGTKGZ  1$1199M(oo--GO6@++%%CK-N^12/Fb D #J MM))EM,, ))11I ''//H:MM))EM  ^^++FNNrUrcrtj|}tjt|||<|S)z)Helper function to invert an index array.)r3 zeros_likerWrA)rinvs rJrrs+ -- CyyS"CH JrUc"t|tS)aIs `x` of dia_matrix type? Parameters ---------- x object to check for being a dia matrix Returns ------- bool True if `x` is a dia matrix, False otherwise Examples -------- >>> from scipy.sparse import dia_array, dia_matrix, coo_matrix, isspmatrix_dia >>> isspmatrix_dia(dia_matrix([[5]])) True >>> isspmatrix_dia(dia_array([[5]])) False >>> isspmatrix_dia(coo_matrix([[5]])) False )r1r)rts rJrrs. a $$rUceZdZdZy)ra< Sparse array with DIAgonal storage. This can be instantiated in several ways: dia_array(D) where D is a 2-D ndarray dia_array(S) with another sparse array or matrix S (equivalent to S.todia()) dia_array((M, N), [dtype]) to construct an empty array with shape (M, N), dtype is optional, defaulting to dtype='d'. dia_array((data, offsets), shape=(M, N)) where the ``data[k,:]`` stores the diagonal entries for diagonal ``offsets[k]`` (See example below) Attributes ---------- dtype : dtype Data type of the array shape : 2-tuple Shape of the array ndim : int Number of dimensions (this is always 2) nnz size data DIA format data array of the array offsets DIA format offset array of the array T Notes ----- Sparse arrays can be used in arithmetic operations: they support addition, subtraction, multiplication, division, and matrix power. Sparse arrays with DIAgonal storage do not support slicing. Examples -------- >>> import numpy as np >>> from scipy.sparse import dia_array >>> dia_array((3, 4), dtype=np.int8).toarray() array([[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]], dtype=int8) >>> data = np.array([[1, 2, 3, 4]]).repeat(3, axis=0) >>> offsets = np.array([0, -1, 2]) >>> dia_array((data, offsets), shape=(4, 4)).toarray() array([[1, 0, 3, 0], [1, 2, 0, 4], [0, 2, 3, 0], [0, 0, 3, 4]]) >>> from scipy.sparse import dia_array >>> n = 10 >>> ex = np.ones(n) >>> data = np.array([ex, 2 * ex, ex]) >>> offsets = np.array([-1, 0, 1]) >>> dia_array((data, offsets), shape=(n, n)).toarray() array([[2., 1., 0., ..., 0., 0., 0.], [1., 2., 1., ..., 0., 0., 0.], [0., 1., 2., ..., 0., 0., 0.], ..., [0., 0., 0., ..., 2., 1., 0.], [0., 0., 0., ..., 1., 2., 1.], [0., 0., 0., ..., 0., 1., 2.]]) NrrrrrorUrJrrHrUrceZdZdZy)raO Sparse matrix with DIAgonal storage. This can be instantiated in several ways: dia_matrix(D) where D is a 2-D ndarray dia_matrix(S) with another sparse array or matrix S (equivalent to S.todia()) dia_matrix((M, N), [dtype]) to construct an empty matrix with shape (M, N), dtype is optional, defaulting to dtype='d'. dia_matrix((data, offsets), shape=(M, N)) where the ``data[k,:]`` stores the diagonal entries for diagonal ``offsets[k]`` (See example below) Attributes ---------- dtype : dtype Data type of the matrix shape : 2-tuple Shape of the matrix ndim : int Number of dimensions (this is always 2) nnz size data DIA format data array of the matrix offsets DIA format offset array of the matrix T Notes ----- Sparse matrices can be used in arithmetic operations: they support addition, subtraction, multiplication, division, and matrix power. Sparse matrices with DIAgonal storage do not support slicing. Examples -------- >>> import numpy as np >>> from scipy.sparse import dia_matrix >>> dia_matrix((3, 4), dtype=np.int8).toarray() array([[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]], dtype=int8) >>> data = np.array([[1, 2, 3, 4]]).repeat(3, axis=0) >>> offsets = np.array([0, -1, 2]) >>> dia_matrix((data, offsets), shape=(4, 4)).toarray() array([[1, 0, 3, 0], [1, 2, 0, 4], [0, 2, 3, 0], [0, 0, 3, 4]]) >>> from scipy.sparse import dia_matrix >>> n = 10 >>> ex = np.ones(n) >>> data = np.array([ex, 2 * ex, ex]) >>> offsets = np.array([-1, 0, 1]) >>> dia_matrix((data, offsets), shape=(n, n)).toarray() array([[2., 1., 0., ..., 0., 0., 0.], [1., 2., 1., ..., 0., 0., 0.], [0., 1., 2., ..., 0., 0., 0.], ..., [0., 0., 0., ..., 2., 1., 0.], [0., 0., 0., ..., 1., 2., 1.], [0., 0., 0., ..., 0., 1., 2.]]) NrrorUrJrrDrrUr)$r __docformat____all__numpyr3 _lib._utilrr_matrixr _baser r r r_datar_sputilsrrrrrrrr _sparsetoolsrrrrrrrrrrorUrJrsv% 7577IH, ,D%6I 7IXI9IrU