ó ʽ÷Xc@`s†dZddlmZmZmZddgZddlmZmZddl m Z ddl m Z d „Z d d „Zd „Zd S(s+ Functions that operate on sparse matrices i(tdivisiontprint_functiontabsolute_importt count_blockstestimate_blocksizei(tisspmatrix_csrt csr_matrix(tisspmatrix_csc(tcsr_count_blockscC`stdƒ‚dS(Nsuse .diagonal() instead(tNotImplementedError(tA((s3/tmp/pip-build-7oUkmx/scipy/scipy/sparse/spfuncs.pytextract_diagonal sgffffffæ?c C`sÏt|ƒpt|ƒs't|ƒ}n|jdkr:dSd|koQdknsetdƒ‚nd|d}t|jƒ}|j\}}|ddkrË|ddkrË|dt|dƒ}nd}|d dkr |d dkr |d t|dƒ}nd}||krW||krW|d t|dƒ}||krPdSdSnt|ddkr‘|ddkr‘|d t|dƒ} nd} | |kr§dS||kr·dS||krÇdSdSdS(s—Attempt to determine the blocksize of a sparse matrix Returns a blocksize=(r,c) such that - A.nnz / A.tobsr( (r,c) ).nnz > efficiency iigð?s.efficiency must satisfy 0.0 < efficiency < 1.0g@iigii i$iiN(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(RRRtnnzt ValueErrortfloattshapeR( R t efficiencythigh_efficiencyR tMtNte22te33te66te44((s3/tmp/pip-build-7oUkmx/scipy/scipy/sparse/spfuncs.pyR#s<       cC`s¦|\}}|dks$|dkr3tdƒ‚nt|ƒrm|j\}}t|||||j|jƒSt|ƒrt|j||fƒStt |ƒ|ƒSdS(s]For a given blocksize=(r,c) count the number of occupied blocks in a sparse matrix A isr and c must be positiveN( R RRRtindptrtindicesRRtTR(R t blocksizetrtcRR((s3/tmp/pip-build-7oUkmx/scipy/scipy/sparse/spfuncs.pyRVs   N(t__doc__t __future__RRRt__all__tcsrRRtcscRt _sparsetoolsRR RR(((s3/tmp/pip-build-7oUkmx/scipy/scipy/sparse/spfuncs.pyts   3