如何将SymPy表达式转换为PuLP？添加时触发TypeError错误

错误信息 #

TypeError: Can only add LpConstraintVar, LpConstraint, LpAffineExpression or True objects

原代码 #

import sympy as sp
import pulp


objective = "x+y+z"
constraints = "x + y + z <= 5, x + 3*y - 2*z >= 3"

constraints_list = constraints.split(',')
constraints_list = [constraint.strip() for constraint in constraints_list]

cons = []

for constraint in constraints_list:
    simplified_constraint = sp.simplify(constraint)
    print(simplified_constraint)
    cons.append(simplified_constraint)

simplified_object = sp.simplify(objective)


lp_problem = pulp.LpProblem("My_LP_Problem", pulp.LpMaximize)


x = pulp.LpVariable('x', lowBound=None)
y = pulp.LpVariable('y', lowBound=None)
z = pulp.LpVariable('z', lowBound=None)


lp_problem += simplified_object  #**error here**


for simplified_constraint in cons:
    lp_problem += simplified_constraint  #**error here**
# lp_problem += y + z <=4

# lp_problem.solve()

print("Status:", pulp.LpStatus[lp_problem.status])
print("x =", x.value())
print("y =", y.value())
print("z =", z.value())

疑问说明 #

简化后的约束打印结果为x + y + z <= 5和x + 3*y - 2*z >= 3，但直接添加lp_problem += y + z <=4可正常运行，请问如何将SymPy表达式转换为PuLP兼容格式？

错误原因 #

SymPy处理后的表达式是SymPy自身的对象（如目标函数是Add类型，约束是Rel类型），而PuLP仅接受由自身LpVariable构建的LpAffineExpression或LpConstraint对象，直接添加SymPy对象会触发类型不匹配错误。

转换步骤 #

• 建立符号映射：将SymPy的符号与PuLP定义的变量一一对应，方便替换。
• 转换目标函数：用SymPy的subs()方法将SymPy表达式替换为PuLP变量组合。
• 转换约束：提取SymPy约束的左右表达式和关系运算符，替换为PuLP变量后重新构建约束。

修改后的完整代码 #

import sympy as sp
import pulp

objective = "x+y+z"
constraints = "x + y + z <= 5, x + 3*y - 2*z >= 3"

# 解析并简化约束
constraints_list = [c.strip() for c in constraints.split(',')]
sym_cons = [sp.simplify(c) for c in constraints_list]
sym_objective = sp.simplify(objective)

# 初始化PuLP问题
lp_problem = pulp.LpProblem("My_LP_Problem", pulp.LpMaximize)

# 定义PuLP变量
x = pulp.LpVariable('x', lowBound=None)
y = pulp.LpVariable('y', lowBound=None)
z = pulp.LpVariable('z', lowBound=None)

# 建立SymPy符号到PuLP变量的映射
sym_x, sym_y, sym_z = sp.symbols('x y z')
sym_to_pulp = {sym_x: x, sym_y: y, sym_z: z}

# 转换目标函数并添加到问题
pulp_objective = sym_objective.subs(sym_to_pulp)
lp_problem += pulp_objective

# 转换约束并添加到问题
for sym_constraint in sym_cons:
    lhs = sym_constraint.lhs.subs(sym_to_pulp)
    rhs = sym_constraint.rhs.subs(sym_to_pulp)
    rel_op = sym_constraint.rel_op
    
    if rel_op == '<=':
        lp_problem += lhs <= rhs
    elif rel_op == '>=':
        lp_problem += lhs >= rhs
    elif rel_op == '==':
        lp_problem += lhs == rhs

# 求解并输出结果
lp_problem.solve()
print("Status:", pulp.LpStatus[lp_problem.status])
print("x =", x.value())
print("y =", y.value())
print("z =", z.value())

内容的提问来源于stack exchange，提问作者Minh Trần