使用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中。