Drake中如何获取任意约束的符号形式及实际参数值？

1. 先判断约束的具体类型 #

Drake的约束均继承自Constraint基类，不同约束有对应的子类。用Python的isinstance()即可快速判断类型：

from pydrake.solvers import (
    Constraint, LinearConstraint, QuadraticConstraint, MISO3Constraint
)

# 假设constraint是你拿到的目标约束对象
if isinstance(constraint, LinearConstraint):
    constraint_type = "线性约束"
elif isinstance(constraint, QuadraticConstraint):
    constraint_type = "二次约束"
elif isinstance(constraint, MISO3Constraint):
    constraint_type = "MISO3约束"
else:
    constraint_type = "其他类型约束"
print(f"当前约束类型：{constraint_type}")

2. 提取线性约束的参数与符号形式 #

线性约束的标准形式为 lb ≤ A·x ≤ ub（若仅单侧约束，对应边界为±∞），直接调用约束对象的属性即可获取参数：

• constraint.A()：系数矩阵A
• constraint.lower_bound()：下界向量lb
• constraint.upper_bound()：上界向量ub

示例代码：

if isinstance(constraint, LinearConstraint):
    A = constraint.A()
    lb = constraint.lower_bound()
    ub = constraint.upper_bound()
    print(f"符号形式：{lb} ≤ A·x ≤ {ub}")
    print(f"A矩阵:\n{A}")
    print(f"下界lb: {lb}")
    print(f"上界ub: {ub}")

3. 提取二次约束的参数与符号形式 #

二次约束的标准形式为 lb ≤ 0.5xᵀQx + bᵀx ≤ ub，对应参数获取方式：

• constraint.Q()：二次项系数矩阵Q
• constraint.b()：一次项系数向量b
• constraint.lower_bound()：下界
• constraint.upper_bound()：上界

若你需要的是xᵀQ'x ≤ b形式，可将Q'换算为2*constraint.Q()，常数项对应constraint.upper_bound()即可。

示例代码：

if isinstance(constraint, QuadraticConstraint):
    Q = constraint.Q()
    b = constraint.b()
    lb = constraint.lower_bound()
    ub = constraint.upper_bound()
    print(f"符号形式：{lb} ≤ 0.5xᵀQx + bᵀx ≤ {ub}")
    print(f"Q矩阵:\n{Q}")
    print(f"b向量: {b}")
    print(f"下界lb: {lb}")
    print(f"上界ub: {ub}")

4. 提取MISO3约束的参数与符号形式 #

MISO3约束是混合整数二次规划的专用约束，形式为 xᵀQx + aᵀx + b ≤ 0，参数获取方式：

• constraint.Q()：二次项矩阵Q
• constraint.a()：一次项向量a
• constraint.b()：常数项b
• constraint.variables()：约束涉及的变量索引（可区分连续/整数变量）

示例代码：

if isinstance(constraint, MISO3Constraint):
    Q = constraint.Q()
    a = constraint.a()
    b_val = constraint.b()
    vars_idx = constraint.variables()
    print(f"符号形式：xᵀQx + aᵀx + {b_val} ≤ 0")
    print(f"Q矩阵:\n{Q}")
    print(f"a向量: {a}")
    print(f"常数项b: {b_val}")
    print(f"涉及的变量索引: {vars_idx}")

通用注意事项 #

• 若约束是通过Binding对象包装的（比如从MathematicalProgram中取出的约束），需先调用binding.constraint()获取底层约束对象。
• 对于自定义或特殊约束，可通过print(constraint.__doc__)查看文档，或调用constraint.Eval(x)代入测试值验证参数正确性。

内容的提问来源于stack exchange，提问作者HoliInn