3 6^ @sfdZddlmZmZmZddgZddlmZmZddl m Z ddl m Z d d Z dd dZd dZdS)z+ Functions that operate on sparse matrices )divisionprint_functionabsolute_import count_blocksestimate_blocksize)isspmatrix_csr csr_matrix)isspmatrix_csc)csr_count_blockscCs tddS)Nzuse .diagonal() instead)NotImplementedError)Ar5/tmp/pip-build-vw4w4j08/scipy/scipy/sparse/spfuncs.pyextract_diagonal srffffff?c Csdt|pt|st|}|jdkr&dSd|ko8dknsFtdd|d}t|j}|j\}}|ddkr|ddkr|dt|d}nd}|d dkr|d dkr|d t|d}nd}||ko||kr|d t|d}||krdSdSnb|ddkr.|ddkr.|d t|d} nd} | |kr@dS||krNdS||kr\dSdSdS)zAttempt to determine the blocksize of a sparse matrix Returns a blocksize=(r,c) such that - A.nnz / A.tobsr( (r,c) ).nnz > efficiency rrg?z.efficiency must satisfy 0.0 < efficiency < 1.0g@g $N)rr)rr)rr)rr)rr)rr)rr)rr)rr)rr)rr)rr r nnz ValueErrorfloatshaper) r Z efficiencyZhigh_efficiencyrMNZe22Ze33Ze66Ze44rrrr#s<        cCsr|\}}|dks|dkr tdt|rH|j\}}t|||||j|jSt|r`t|j||fStt ||SdS)z]For a given blocksize=(r,c) count the number of occupied blocks in a sparse matrix A rzr and c must be positiveN) rrrr Zindptrindicesr rTr )r blocksizercrrrrrrVs N)r)__doc__ __future__rrr__all__Zcsrrr Zcscr Z _sparsetoolsr rrrrrrrs   3