B &]\ @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)Ar3lib/python3.7/site-packages/scipy/sparse/spfuncs.pyextract_diagonal srffffff?c Cs^t|st|st|}|jdkr&dSd|kr:dksDntdd|d}t|j}|j\}}|ddkr|ddkr|dt|d}nd }|d dkr|d dkr|d t|d }nd }||kr||kr|d t|d}||krdSd Snb|ddkr(|ddkr(|dt|d} nd } | |kr:dS||krHd S||krVdSdSdS)zAttempt to determine the blocksize of a sparse matrix Returns a blocksize=(r,c) such that - A.nnz / A.tobsr( (r,c) ).nnz > efficiency r)rrg?z.efficiency must satisfy 0.0 < efficiency < 1.0g@)rrg )rr$)r)rrN)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 Z blocksizercrrrrrrVs N)r)__doc__Z __future__rrr__all__Zcsrrr Zcscr Z _sparsetoolsr rrrrrrrs   3