ó î&]\c@`sKdZddlmZmZmZdZddgZddlZddlZddl Z ddl Z ddl m ZmZmZmZmZdd lmZmZdd lmZmZmZmZmZmZmZmZmZm Z ydd lm!Z"Wne#k rd „Z"nXdeee$fd „ƒYZ%d„Z&d„Z'dS(sDictionary Of Keys based matrixi(tdivisiontprint_functiontabsolute_importsrestructuredtext ent dok_matrixtisspmatrix_dokN(tziptxranget iteritemstiterkeyst itervaluesi(tspmatrixt isspmatrix( tisdensetgetdtypetisshapet isintliket isscalarliketupcastt upcast_scalart IndexMixintget_index_dtypet check_shape(tisSequenceTypecC`s+t|dƒp*t|dƒp*t|dƒS(Nt__len__t__next__tnext(thasattr(tx((s/lib/python2.7/site-packages/scipy/sparse/dok.pyt _is_sequencescB`sÐeZdZdZd#d#ed„Zd„Zd„Zd„Z e de j de ƒZ d#d„Zd „Ze jje_e jje_d „Zd d „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd#ed„Ze jje_d„Z d„Z!e j!je!_d„Z"d„Z#ed„Z$e j$je$_ed „Z%e j%je%_ed!„Z&e j&je&_d"„Z'e j'je'_RS($sð Dictionary Of Keys based sparse matrix. This is an efficient structure for constructing sparse matrices incrementally. This can be instantiated in several ways: dok_matrix(D) with a dense matrix, D dok_matrix(S) with a sparse matrix, S dok_matrix((M,N), [dtype]) create the matrix with initial shape (M,N) dtype is optional, defaulting to dtype='d' 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 Number of nonzero elements Notes ----- Sparse matrices can be used in arithmetic operations: they support addition, subtraction, multiplication, division, and matrix power. Allows for efficient O(1) access of individual elements. Duplicates are not allowed. Can be efficiently converted to a coo_matrix once constructed. Examples -------- >>> import numpy as np >>> from scipy.sparse import dok_matrix >>> S = dok_matrix((5, 5), dtype=np.float32) >>> for i in range(5): ... for j in range(5): ... S[i, j] = i + j # Update element tdokc C`s§tj|ƒtj|ƒt|dtƒ|_t|tƒrnt|ƒrn|\}}t ||fƒ|_ n5t |ƒröt |ƒr›|r›|j ƒ}n |jƒ}|dk rÅ|j|ƒ}ntj||ƒt |jƒ|_ |j|_n­ytj|ƒ}Wntk r(tdƒ‚nXt|jƒdkrMtdƒ‚nddlm}||d|ƒjƒ}tj||ƒt |jƒ|_ |j|_dS(NtdefaultsInvalid input format.is(Expected rank <=2 dense array or matrix.i(t coo_matrixtdtype(tdictt__init__R R tfloatR t isinstancettupleRRt_shapeR RtcopyttodoktNonetastypetupdatetshapetnptasarrayt Exceptiont TypeErrortlentcooR( tselftarg1R,R R'tMtNRtd((s/lib/python2.7/site-packages/scipy/sparse/dok.pyR"Qs4       cC`stdƒ‚dS(Ns9Direct modification to dok_matrix element is not allowed.(tNotImplementedError(R3tval((s/lib/python2.7/site-packages/scipy/sparse/dok.pyR+tscC`stj||ƒS(sÒAn update method for dict data defined for direct access to `dok_matrix` data. Main purpose is to be used for effcient conversion from other spmatrix classes. Has no checking if `data` is valid.(R!R+(R3tdata((s/lib/python2.7/site-packages/scipy/sparse/dok.pyt_updateyscC`sN|j|dtƒj|jƒ}|j|_tj|ƒtj||ƒdS(NR'(treshapetFalsetasformattformatt__dict__R!tclearR+(R3R,t new_matrix((s/lib/python2.7/site-packages/scipy/sparse/dok.pyt set_shapes!  tfgettfsetcC`s(|dk rtdƒ‚ntj|ƒS(Ns6getnnz over an axis is not implemented for DOK format.(R)R8R!R(R3taxis((s/lib/python2.7/site-packages/scipy/sparse/dok.pytgetnnz‡s cC`std„t|ƒDƒƒS(Ncs`s|]}|dkVqdS(iN((t.0R((s/lib/python2.7/site-packages/scipy/sparse/dok.pys Žs(tsumR (R3((s/lib/python2.7/site-packages/scipy/sparse/dok.pyt count_nonzeroscC`s tj|ƒS(N(R!R(R3((s/lib/python2.7/site-packages/scipy/sparse/dok.pyR“sgcC`s·y.|\}}t|ƒr't|ƒs-t‚Wn&tttfk rVtdƒ‚nX|dks•||jdks•|dks•||jdkr¤tdƒ‚ntj|||ƒS(stThis overrides the dict.get method, providing type checking but otherwise equivalent functionality. s!Index must be a pair of integers.iisIndex out of bounds.(RtAssertionErrorR0t ValueErrort IndexErrorR,R!tget(R3tkeyRtitj((s/lib/python2.7/site-packages/scipy/sparse/dok.pyRN–s ">cC`s9|jjdƒ}|j|ƒ\}}t|ƒ}t|ƒ}|r|rt|ƒ}t|ƒ}|dkrƒ||jd7}n|dks¢||jdkr±tdƒ‚n|dkrÑ||jd7}n|dksð||jdkrÿtdƒ‚ntj|||f|ƒS|s-t |t ƒr |sBt |t ƒr |r[t ||dƒn|}|rzt ||dƒn|}|j |jdƒ} |j |jdƒ} t | Œ} t | Œ} t | ƒt | ƒf} t| ƒ}t |ƒd|kr |dkr |j| | | ƒSn|j||ƒ\}}|jdkr]t|jd|jƒS|jƒ}||jd ks–|jƒ|jdkrÊtd|jƒ|jd|jddfƒ‚n|dkr|jƒ}||dkc|jd7Ks(RRR RR,R]R^RR!RNRRLRR+R-terrstateRR ttocscR ttodensetNotImplemented(R3RŠt res_dtypeR5R6ROtaijtcsc((R‰RŠs/lib/python2.7/site-packages/scipy/sparse/dok.pyt__add__6s0 %  )    c`sXtˆƒr‹tˆjdˆjƒ}ˆj\}}xtjt|ƒt|ƒƒD]2}tjˆ|dƒˆ}|rR|||cs(RRR,R R]R^RR!RNRRLR+RR RŒR RRŽ(R3RŠR‰R5R6RORR‘((RŠR3s/lib/python2.7/site-packages/scipy/sparse/dok.pyt__radd__Us* %  #    c`scˆjjdkr!tdƒ‚ntˆjdˆjƒ}tj|‡fd†tˆƒDƒƒ|S(NR}s2Negating a sparse boolean matrix is not supported.R c3`s |]}|ˆ| fVqdS(N((RHRˆ(R3(s/lib/python2.7/site-packages/scipy/sparse/dok.pys rs(R tkindR8RR,R!R+R(R3R‰((R3s/lib/python2.7/site-packages/scipy/sparse/dok.pyt__neg__ms &c`sQt|jˆƒ}t|jd|ƒ}tj|‡fd†t|ƒDƒƒ|S(NR c3`s%|]\}}||ˆfVqdS(N((RHRˆRo(RŠ(s/lib/python2.7/site-packages/scipy/sparse/dok.pys ys(RR RR,R!R+R(R3RŠRR‰((RŠs/lib/python2.7/site-packages/scipy/sparse/dok.pyt _mul_scalarus&cC`sjtj|jddt|j|jƒƒ}x8t|ƒD]*\\}}}||c|||7Žs(RR!R+RRŽ(R3RŠ((RŠs/lib/python2.7/site-packages/scipy/sparse/dok.pyt__imul__Œs &c`sktˆƒr]t|jˆƒ}t|jd|ƒ}tj|‡fd†t|ƒDƒƒ|S|jƒˆS(NR c3`s%|]\}}||ˆfVqdS(N((RHRˆRo(RŠ(s/lib/python2.7/site-packages/scipy/sparse/dok.pys –s( RRR RR,R!R+Rttocsr(R3RŠRR‰((RŠs/lib/python2.7/site-packages/scipy/sparse/dok.pyt __truediv__’s  &c`s:tˆƒr6tj|‡fd†t|ƒDƒƒ|StS(Nc3`s%|]\}}||ˆfVqdS(N((RHRˆRo(RŠ(s/lib/python2.7/site-packages/scipy/sparse/dok.pys œs(RR!R+RRŽ(R3RŠ((RŠs/lib/python2.7/site-packages/scipy/sparse/dok.pyt __itruediv__šs &cC`s tj|ƒS(N(R!t __reduce__(R3((s/lib/python2.7/site-packages/scipy/sparse/dok.pyR¡ scC`so|dk rtdƒ‚n|j\}}t||fd|jd|ƒ}tj|d„t|ƒDƒƒ|S(NsoSparse matrices do not support an 'axes' parameter because swapping dimensions is the only logical permutation.R R'cs`s-|]#\\}}}||f|fVqdS(N((RHtlefttrightR9((s/lib/python2.7/site-packages/scipy/sparse/dok.pys ²s(R)RLR,RR R!R+R(R3taxesR'R5R6R‰((s/lib/python2.7/site-packages/scipy/sparse/dok.pyt transposeªs !cC`sN|j\}}t||fd|jƒ}tj|d„t|ƒDƒƒ|S(sReturn the conjugate transpose.R cs`s6|],\\}}}||ftj|ƒfVqdS(N(R-tconj(RHR¢R£R9((s/lib/python2.7/site-packages/scipy/sparse/dok.pys ¼s(R,RR R!R+R(R3R5R6R‰((s/lib/python2.7/site-packages/scipy/sparse/dok.pyt conjtransp¸s cC`s,t|jd|jƒ}tj||ƒ|S(NR (RR,R R!R+(R3R‰((s/lib/python2.7/site-packages/scipy/sparse/dok.pyR'Àsc`sVtdˆjdfdˆjƒ}tj|‡‡fd†tˆjdƒDƒƒ|S(s-Returns the i-th row as a (1 x n) DOK matrix.iR c3`s+|]!}d|fˆˆ|ffVqdS(iN((RHRQ(RPR3(s/lib/python2.7/site-packages/scipy/sparse/dok.pys Ês(RR,R R!R+R(R3RPR‰((RPR3s/lib/python2.7/site-packages/scipy/sparse/dok.pytgetrowÇs"0c`sVtˆjddfdˆjƒ}tj|‡‡fd†tˆjdƒDƒƒ|S(s0Returns the j-th column as a (m x 1) DOK matrix.iiR c3`s+|]!}|dfˆ|ˆffVqdS(iN((RHRP(RQR3(s/lib/python2.7/site-packages/scipy/sparse/dok.pys Ðs(RR,R R!R+R(R3RQR‰((RQR3s/lib/python2.7/site-packages/scipy/sparse/dok.pytgetcolÍs"0cC`sddlm}|jdkr5||jd|jƒStdt|jƒƒ}tjt |ƒd|jd|jƒ}tjd„t |ƒDƒd|d|jƒ}tjd„t |ƒDƒd|d|jƒ}||||ffd |jd|jƒ}t |_ |S( Ni(RiR tmaxvaltcountcs`s|]\}}|VqdS(N((RHRPR„((s/lib/python2.7/site-packages/scipy/sparse/dok.pys Úscs`s|]\}}|VqdS(N((RHR„RQ((s/lib/python2.7/site-packages/scipy/sparse/dok.pys ÛsR,( R2RtnnzR,R RR\R-tfromiterR RtTruethas_canonical_format(R3R'Rt idx_dtypeR:trowtcoltA((s/lib/python2.7/site-packages/scipy/sparse/dok.pyttocooÓs'..* cC`s|r|jƒS|S(N(R'(R3R'((s/lib/python2.7/site-packages/scipy/sparse/dok.pyR(âs cC`s|jdtƒjd|ƒS(NR'(R´R=RŒ(R3R'((s/lib/python2.7/site-packages/scipy/sparse/dok.pyRŒéscG`sšt|ƒ}|\}}|j\}}||ks?||krxKtt|ƒƒD]4\}}||ksv||krR|||f=qRqRWn||_dS(N(RR,tlistRR&(R3R,tnewMtnewNR5R6RPRQ((s/lib/python2.7/site-packages/scipy/sparse/dok.pytresizeîs  N((t__name__t __module__t__doc__R?R)R=R"R+R;RCtpropertyR t get_shapeR,RGRJRRNRpRXR_R’R“R•R–R™RœRRŸR R¡R¥R§R'R¨R©R´R(RŒR¸(((s/lib/python2.7/site-packages/scipy/sparse/dok.pyRsR0#       J  4                  cC`s t|tƒS(sÍIs x of dok_matrix type? Parameters ---------- x object to check for being a dok matrix Returns ------- bool True if x is a dok matrix, False otherwise Examples -------- >>> from scipy.sparse import dok_matrix, isspmatrix_dok >>> isspmatrix_dok(dok_matrix([[5]])) True >>> from scipy.sparse import dok_matrix, csr_matrix, isspmatrix_dok >>> isspmatrix_dok(csr_matrix([[5]])) False (R$R(R((s/lib/python2.7/site-packages/scipy/sparse/dok.pyRüscC`s)t|ƒdkrdStjtj|ƒS(sFProduct of a list of numbers; ~40x faster vs np.prod for Python tuplesii(R1t functoolstreducetoperatortmul(R((s/lib/python2.7/site-packages/scipy/sparse/dok.pyRWs((R»t __future__RRRt __docformat__t__all__R¾RÀR]tnumpyR-tscipy._lib.sixRR‚RRRR tbaseR R tsputilsR R RRRRRRRRRRt ImportErrorR!RRRW(((s/lib/python2.7/site-packages/scipy/sparse/dok.pyts$     (F  ÿß