python - 使用 sympy 将相似的术语放在一起

Question

以下代码将类似的术语放在一起。然而，对于大量术语，将所有相似术语相加可能会变得更加困难。关于如何在不使用我目前采用的繁琐方法的情况下实现这一目标的任何建议。

import sympy as sp
from sympy import *
import numpy as np
from numpy.linalg import *
x,y,a,b,c,d=sp.symbols('x,y,a,b,c,d')
A=a*x + b*y
B=c*x+d*y
A1=A.coeff(x)
B1=B.coeff(x)                    #segregating the coefficients
A2=A.coeff(y)
B2=B.coeff(y)
A_new=(A1+B1)*x + (A2+B2)*y
print A_new

输出为

x*(a + c) + y*(b + d)

Answer 1

import sympy as sp

x,y,a,b,c,d=sp.symbols('x,y,a,b,c,d')
A = a*x + b*y
B = c*x+d*y
sum_expr = A+B
sum_expr.collect((x,y))  # output: x*(a + c) + y*(b + d)

写在 the tutorial也是。