o o hr@shddlmZddlmZddlmZddlmZddlm Z m Z m Z GdddeZ Gd d d eZ d S) )S)_sympify)KroneckerDelta) MatrixExpr) ZeroMatrixIdentity OneMatrixcspeZdZdZfddZeddZeddZdd Zd d Z d d Z ddZ e Z Z ddZddZZS)PermutationMatrixaA Permutation Matrix Parameters ========== perm : Permutation The permutation the matrix uses. The size of the permutation determines the matrix size. See the documentation of :class:`sympy.combinatorics.permutations.Permutation` for the further information of how to create a permutation object. Examples ======== >>> from sympy import Matrix, PermutationMatrix >>> from sympy.combinatorics import Permutation Creating a permutation matrix: >>> p = Permutation(1, 2, 0) >>> P = PermutationMatrix(p) >>> P = P.as_explicit() >>> P Matrix([ [0, 1, 0], [0, 0, 1], [1, 0, 0]]) Permuting a matrix row and column: >>> M = Matrix([0, 1, 2]) >>> Matrix(P*M) Matrix([ [1], [2], [0]]) >>> Matrix(M.T*P) Matrix([[2, 0, 1]]) See Also ======== sympy.combinatorics.permutations.Permutation cs:ddlm}t|}t||std|t||S)Nr Permutationz({} must be a SymPy Permutation instance.) sympy.combinatorics.permutationsr r isinstance ValueErrorformatsuper__new__)clspermr  __class__z/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/matrices/expressions/permutation.pyr;s  zPermutationMatrix.__new__cCs|jdj}||fSNr)argssize)selfrrrrshapeEs zPermutationMatrix.shapecC |jdjSr)r is_IdentityrrrrrJ zPermutationMatrix.is_IdentitycKs|jrt|jS|S)N)rrrows)rhintsrrrdoitNs zPermutationMatrix.doitcKs|jd}t|||Sr)rrapply)rijkwargsrrrr_entrySs zPermutationMatrix._entrycCst|jd|Sr)r rr$)rexprrr _eval_powerWszPermutationMatrix._eval_powercCst|jddS)Nr)r rr rrr _eval_inverseZszPermutationMatrix._eval_inversecCs.|jd}|dkrtjS|dkrtjSt)Nrrr,)r signaturerOne NegativeOneNotImplementedError)rsignrrr_eval_determinant_s z#PermutationMatrix._eval_determinantcsddlm}ddlm}|jd}|j}g}d\}} } d} |D]x} t| } t| }| sH|d|| kr=d} | g}|} | } q|| g|| 7}q|| krs|d|| | krg|| ||d} |d}q|} || | | 7} q| d|| | kr|| ||d} | d}q|| | | 7} qdg}|D]0}g}d} |D]} fdd | D}||| t| 7} q| 7||}t |}||q||S) Nrr r)BlockDiagMatrix)rrrFTcsg|]}|qSrr).0r&prr szFPermutationMatrix._eval_rewrite_as_BlockDiagMatrix..) r r blockmatrixr4rfull_cyclic_formlenmaxappendr )rrr(r r4rr: cycles_picksabcflagcyclelmtemppick new_cycles new_cyclematrr6r _eval_rewrite_as_BlockDiagMatrixgs`                  z2PermutationMatrix._eval_rewrite_as_BlockDiagMatrix)__name__ __module__ __qualname____doc__rpropertyrrr$r)r+r-_eval_transpose _eval_adjointr3rK __classcell__rrrrr s 1  r csLeZdZdZejffdd Zd ddZeddZ d d Z d d Z Z S) MatrixPermuteazSymbolic representation for permuting matrix rows or columns. Parameters ========== perm : Permutation, PermutationMatrix The permutation to use for permuting the matrix. The permutation can be resized to the suitable one, axis : 0 or 1 The axis to permute alongside. If `0`, it will permute the matrix rows. If `1`, it will permute the matrix columns. Notes ===== This follows the same notation used in :meth:`sympy.matrices.matrixbase.MatrixBase.permute`. Examples ======== >>> from sympy import Matrix, MatrixPermute >>> from sympy.combinatorics import Permutation Permuting the matrix rows: >>> p = Permutation(1, 2, 0) >>> A = Matrix([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) >>> B = MatrixPermute(A, p, axis=0) >>> B.as_explicit() Matrix([ [4, 5, 6], [7, 8, 9], [1, 2, 3]]) Permuting the matrix columns: >>> B = MatrixPermute(A, p, axis=1) >>> B.as_explicit() Matrix([ [2, 3, 1], [5, 6, 4], [8, 9, 7]]) See Also ======== sympy.matrices.matrixbase.MatrixBase.permute c sddlm}t|}|jstd|t|}t|tr"|jd}t||s.td|t|}|dvr:td|j |}||j kr\z| |}Wnty[td|||wt ||||S)Nrr z#{} must be a SymPy matrix instance.z>{} must be a SymPy Permutation or a PermutationMatrix instance)rrzThe axis must be 0 or 1.zsSize does not match between the permutation {} and the matrix {} threaded over the axis {} and cannot be converted.)r r r is_Matrixrrrr rrrresizerr)rrJraxisr mat_sizerrrrs8        zMatrixPermute.__new__TcKs|j\}}}|r|jdd|i|}|jdd|i|}|jr!|S|jr8|tjur-t|S|tjur8t|dSt|tt frA|St|t r[|jd|kr[t |jd||jd|S|S)Ndeepr,rrr) rr$rrZeror r/rrr rT)rrYr#rJrrWrrrr$s    zMatrixPermute.doitcCrr)rrr rrrrr!zMatrixPermute.shapecKsD|j\}}}|dkr||||fS|dkr ||||fSdS)Nrr)rr%)rr&r'r(rJrrWrrrr)s zMatrixPermute._entrycOsfddlm}|j\}}}|dd}|r||}|dkr$|t||S|dkr1||t|dSdS)Nr)MatMulrYTrr,)matmulr\rgetrewriter )rrr(r\rJrrWrYrrr_eval_rewrite_as_MatMul"s    z%MatrixPermute._eval_rewrite_as_MatMul)T) rLrMrNrOrr[rr$rPrr)r`rSrrrrrTs3 " rTN) sympy.corersympy.core.sympifyrsympy.functionsrmatexprrspecialrrr r rTrrrrs