在 sympy 中显示一个公因数矩阵

Question

我想显示一个矩阵，在 sympy 中计算矩阵后，将提取的公因子放在矩阵的外部。

我写了下面的代码。

from sympy import *
a = symbols("a")
b = symbols("b")

A = Matrix([exp(I*a),exp(I*a)*exp(I*b)])
print simplify(A)

我低于输出。

Matrix([
[      exp(I*a)],
[exp(I*(a + b))]])

但是，我想要低于输出。

exp(I*a)*Matrix([
[      1],
[exp(I*b)]])

我尝试了 collect(A,exp(I*a)) 并得到跟随错误。

Traceback (most recent call last):

  File "<ipython-input-65-834f4c326df4>", line 1, in <module>
    runfile('C:/Anaconda2/Programs/test/untitled44.py', wdir='C:/Anaconda2/Programs/test')

  File "C:\Anaconda2\lib\site-packages\spyderlib\widgets\externalshell\sitecustomize.py", line 699, in runfile
    execfile(filename, namespace)

  File "C:\Anaconda2\lib\site-packages\spyderlib\widgets\externalshell\sitecustomize.py", line 74, in execfile
    exec(compile(scripttext, filename, 'exec'), glob, loc)

  File "C:/Anaconda2/Programs/test/untitled44.py", line 14, in <module>
    collect(A,exp(I*a))

  File "C:\Anaconda2\lib\site-packages\sympy\simplify\simplify.py", line 451, in collect
    if expr.is_Mul:

  File "C:\Anaconda2\lib\site-packages\sympy\matrices\matrices.py", line 3084, in __getattr__
    "%s has no attribute %s." % (self.__class__.__name__, attr))

AttributeError: MutableDenseMatrix has no attribute is_Mul.

我知道一种提取矩阵元素公因数的方法，如下所示 link。 https://github.com/sympy/sympy/issues/8442

但这不是我的愿望。

我该怎么办？

Answer 1

我认为Sympy 没有为您想要的任务提供功能。但是，您可以按照 Mathematica SE (link) 中提出的类似问题的已接受答案中提出的方法手动执行此操作。

想法是通过gcd提取多项式元素的公因子，然后使用MatMul和evaluate=False选项来限制Sympy执行标量矩阵乘法。

import sympy as sp

a, b = sp.symbols('a, b')

A = sp.Matrix([sp.exp(sp.I * a), sp.exp(sp.I * a) * sp.exp(sp.I * b)])
g = sp.gcd(tuple(A))

A_v2 = sp.MatMul(g,(A/g),evaluate = False)
print(A_v2)

exp(I*a)*Matrix([
[       1],
[exp(I*b)]])

在 sympy 中显示一个公因数矩阵

Display a matrix with putting a common factor in sympy

python

matrix

symbolic-math

sympy