B ˜‘[Þã@s<ddlmZmZddlmZddlmZdd„Zdd„ZdS) é)Úprint_functionÚdivision)Úrange)Ú SparseMatrixcCs¢dd„t| ¡ŽDƒ\}}}dg|r,|dndd}x0t|ƒD]$\}}| |g|||d¡q@W| t|ƒg|jt|ƒd¡|j|jg}||||gS)aÇConverts a sparse matrix to Compressed Sparse Row (CSR) format. Parameters ========== A : contains non-zero elements sorted by key (row, column) JA : JA[i] is the column corresponding to A[i] IA : IA[i] contains the index in A for the first non-zero element of row[i]. Thus IA[i+1] - IA[i] gives number of non-zero elements row[i]. The length of IA is always 1 more than the number of rows in the matrix. cSsg|] }t|ƒ‘qS©)Úlist)Ú.0Úirrú9lib/python3.7/site-packages/sympy/matrices/sparsetools.pyú sz_doktocsr..ré)ÚzipZrow_listÚ enumerateÚextendÚlenZrowsZcols)ZdokÚrowÚJAÚAÚIAr ÚrÚshaperrr Ú _doktocsrs  " rc Cszi}|\}}}}xZtt|ƒdƒD]F}t||||dƒ}x*t||||ƒD]\}} |||| f<qPWq"Wt||gŽS)z3Converts a CSR representation to DOK representationr )rrÚslicer r) ZcsrZsmatrrrrr ÚindicesÚlÚmrrr Ú _csrtodoks rN) Z __future__rrZsympy.core.compatibilityrZsympyrrrrrrr Ús