wg#VddlmZddlmZddlmZddlmZmZddl m Z dZ dZ d Zy ) )Dict)Dummy) is_sequence)as_int filldedent)MutableSparseMatrixct|jDcgc] }t|c}\}}}dg|r|dnddzz}t|D]#\}}|j |g|||dz z z%|j t |g|j t |z dzz|j |jg}||||gScc}w)aConverts 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. Examples ======== >>> from sympy.matrices.sparsetools import _doktocsr >>> from sympy import SparseMatrix, diag >>> m = SparseMatrix(diag(1, 2, 3)) >>> m[2, 0] = -1 >>> _doktocsr(m) [[1, 2, -1, 3], [0, 1, 0, 2], [0, 1, 2, 4], [3, 3]] rr)ziprow_listlist enumerateextendlenrowscols)dokirowJAAIArshapes _/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/sympy/matrices/sparsetools.py _doktocsrr s0$' #78a$q'8JCQ s1v!q( )B#(1 1#q3q1u:~&'(IIs1vh3r7*Q./0 XXsxx E r2u  9sCci}|\}}}}tt|dz D]8}t||||dz}t||||D] \}} |||| f<:t g||S)a"Converts a CSR representation to DOK representation. Examples ======== >>> from sympy.matrices.sparsetools import _csrtodok >>> _csrtodok([[5, 8, 3, 6], [0, 1, 2, 1], [0, 0, 2, 3, 4], [4, 3]]) Matrix([ [0, 0, 0], [5, 8, 0], [0, 0, 3], [0, 6, 0]]) r)rangerslicer SparseMatrix) csrsmatrrrrrindiceslms r _csrtodokr&*s DAr2u 3r7Q; 1r!a%y)' BwK0 DAqDAJ   % % %%c t|dvrtt|dttfstt|dk(r?|j dd}|j dd}| t |}|At |}n5t|dk(rt |dx}}ntt |dd\}}td |dD}d }ig}t|djD]\}}||\} } t|ttfrgd} t|D]V\} | | z } t!r4t#| | z| | zf<| t%j&dz z } J| | z| | zf<Xt!|rt#|}|j&\} }|r9|r7t)|| z | \}}t)|| z |\}}|xs|}t%||}n.|rt)|| z | \}}n|rt)|| z |\}}nd}d}|rttd t%|j&}t+|D]} || | f<| |z } | |z } k|so| | f<|j-||ft#d}|j/|*||j0krtd |d|j0|*||j2krtd|d|j2||cxurnn|j0}|j2}n5||t5||j2}n||t5||j0}fd}|rh|D]c\}||\} } t7rnfd}d} | | z|ks.| | z|ks7|| | z| | z|| | dz } | | z|ks[| | z|kr-et#||S#ttf$rttd wxYw)aWReturns a SparseMatrix from the given dictionary describing the diagonals of the matrix. The keys are positive for upper diagonals and negative for those below the main diagonal. The values may be: * expressions or single-argument functions, * lists or tuples of values, * matrices Unless dimensions are given, the size of the returned matrix will be large enough to contain the largest non-zero value provided. kwargs ====== rows : rows of the resulting matrix; computed if not given. cols : columns of the resulting matrix; computed if not given. Examples ======== >>> from sympy import banded, ones, Matrix >>> from sympy.abc import x If explicit values are given in tuples, the matrix will autosize to contain all values, otherwise a single value is filled onto the entire diagonal: >>> banded({1: (1, 2, 3), -1: (4, 5, 6), 0: x}) Matrix([ [x, 1, 0, 0], [4, x, 2, 0], [0, 5, x, 3], [0, 0, 6, x]]) A function accepting a single argument can be used to fill the diagonal as a function of diagonal index (which starts at 0). The size (or shape) of the matrix must be given to obtain more than a 1x1 matrix: >>> s = lambda d: (1 + d)**2 >>> banded(5, {0: s, 2: s, -2: 2}) Matrix([ [1, 0, 1, 0, 0], [0, 4, 0, 4, 0], [2, 0, 9, 0, 9], [0, 2, 0, 16, 0], [0, 0, 2, 0, 25]]) The diagonal of matrices placed on a diagonal will coincide with the indicated diagonal: >>> vert = Matrix([1, 2, 3]) >>> banded({0: vert}, cols=3) Matrix([ [1, 0, 0], [2, 1, 0], [3, 2, 1], [0, 3, 2], [0, 0, 3]]) >>> banded(4, {0: ones(2)}) Matrix([ [1, 1, 0, 0], [1, 1, 0, 0], [0, 0, 1, 1], [0, 0, 1, 1]]) Errors are raised if the designated size will not hold all values an integral number of times. Here, the rows are designated as odd (but an even number is required to hold the off-diagonal 2x2 ones): >>> banded({0: 2, 1: ones(2)}, rows=5) Traceback (most recent call last): ... ValueError: sequence does not fit an integral number of times in the matrix And here, an even number of rows is given...but the square matrix has an even number of columns, too. As we saw in the previous example, an odd number is required: >>> banded(4, {0: 2, 1: ones(2)}) # trying to make 4x4 and cols must be odd Traceback (most recent call last): ... ValueError: sequence does not fit an integral number of times in the matrix A way around having to count rows is to enclosing matrix elements in a tuple and indicate the desired number of them to the right: >>> banded({0: 2, 2: (ones(2),)*3}) Matrix([ [2, 0, 1, 1, 0, 0, 0, 0], [0, 2, 1, 1, 0, 0, 0, 0], [0, 0, 2, 0, 1, 1, 0, 0], [0, 0, 0, 2, 1, 1, 0, 0], [0, 0, 0, 0, 2, 0, 1, 1], [0, 0, 0, 0, 0, 2, 1, 1]]) An error will be raised if more than one value is written to a given entry. Here, the ones overlap with the main diagonal if they are placed on the first diagonal: >>> banded({0: (2,)*5, 1: (ones(2),)*3}) Traceback (most recent call last): ... ValueError: collision at (1, 1) By placing a 0 at the bottom left of the 2x2 matrix of ones, the collision is avoided: >>> u2 = Matrix([ ... [1, 1], ... [0, 1]]) >>> banded({0: [2]*5, 1: [u2]*3}) Matrix([ [2, 1, 1, 0, 0, 0, 0], [0, 2, 1, 0, 0, 0, 0], [0, 0, 2, 1, 1, 0, 0], [0, 0, 0, 2, 1, 0, 0], [0, 0, 0, 0, 2, 1, 1], [0, 0, 0, 0, 0, 0, 1]]) )rrrNrr)rc32K|]}t|ywN)r).0ks r zbanded..s,aq ,szNunrecognized input to banded: expecting [[row,] col,] {int: value}c*|dkr| nd}|rdn|}||fS)Nr)drcs rrczbanded..rcs#a%QBQA!t r'zh sequence does not fit an integral number of times in the matrixzDesignated rows z < needed zDesignated cols c\|r)||fvr||f|fvrtd||f|||f<yy)Nz collision at ) ValueError)rjvr"tbas rupdatezbanded..updatesJ 1v~$q!t*S!H"< q!f!>??DAJ r'cSr-r2)_vis rzbanded..'s"r')r TypeError isinstancedictrgetrmapallr7rritemsr tuplerrr minrdivmodrappendtodokrrmaxcallable)argskwargsrrr=r5undoner3r9rr4extrarrvcvnrxrncxcxdor8sr;r"r:r>s @@@rbandedr[BsH6 t9I %O$r(T4L1O t9>::fd+D::fd+Dd|d| Y!^ a/ )D4VT"1X.JD$ ,48, ,  D F 'CR )"1!u1 a$ 'E"1 ,2U r?%b)B)+DQA&S]Q..E)+DQA& ,^QAWWFBq"-Bq"-BH"R[tax,Atax,A -%"&''AGG A2Y QT QQ DAJ MM1a& !S)"T T4 A 779D D166MT166JKK D166MT166JKK tvvvv dl4  dl4  EAra5DAqrl AAa%$,1q54<q1ua!eQqT*Qa%$,1q54<   dD ))i  "6  456 66s C O$O+N)sympy.core.containersrsympy.core.symbolrsympy.utilities.iterablesrsympy.utilities.miscrrsparser r rr&r[r2r'rras%&#137B&0j*r'