3 h @sRdZddgZddlmZmZddlmZddlmZddZ d d dZ d dZ d S)z+ Functions that operate on sparse matrices count_blocksestimate_blocksize)isspmatrix_csr csr_matrix)isspmatrix_csc)csr_count_blockscCs tddS)Nzuse .diagonal() instead)NotImplementedError)Ar 5/tmp/pip-build-riy7u7_k/scipy/scipy/sparse/spfuncs.pyextract_diagonal sr ffffff?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 rg?z.efficiency must satisfy 0.0 < efficiency < 1.0g@g $N)rr)rr)rr)rr)rr)rr)rr)rr)rr)rr)rr)rrrnnz ValueErrorfloatshaper) r Z efficiencyZhigh_efficiencyrMNZe22Ze33Ze66Ze44r r r r!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) rrrrZindptrindicesrrTr)r blocksizercrrr r r rTs N)r ) __doc____all__ZcsrrrZcscrZ _sparsetoolsrr rrr r r r s   3