Python(OR-Tools)中如何检视模型已定义的约束?
解答
完全有方法检视OR-Tools的CP-SAT模型里定义的全部约束,你说的那段循环加行全不同约束的代码跑的时候,每执行一次AddAllDifferent就会在model里生成一个实实在在的、可观测的约束实体,循环跑多少次就生成多少个,不会加完就找不到。
具体说明
- 你举的数独例子里,遍历9行加约束的逻辑跑完,模型里会实打实多9个独立的AllDifferent约束,每个对应一行的单元格变量,不是什么临时的、不可见的中间状态。
- 模型本身带了现成的检视接口,不用自己做额外记录:
- 调用
model.ConstraintsCount()可以直接拿到当前模型里的约束总数,你在加那段循环前后各打一次这个值,就能看到数刚好涨了9,和循环次数完全对得上。 - 直接遍历
model.constraints这个属性就能拿到每一个约束对象,每个对象都能查类型和具体参数:比如判断一个约束是不是全不同约束,就看constraint.HasField("all_diff")是不是返回True,要查这个约束管了哪些变量,直接读constraint.all_diff.exprs就行。其他类型的约束,比如线性等式/不等式、区间约束、调度类的累积约束、允许赋值约束等等,都有对应的字段可以读全量配置,没有黑盒。
- 调用
你可以在那段循环代码后面加几行测试代码跑一下,一眼就能看到结果:
# 加完行AllDifferent约束之后执行 print(f"当前模型总约束数:{model.ConstraintsCount()}") for idx, ct in enumerate(model.constraints): if ct.HasField("all_diff"): print(f"约束{idx}:全不同类型,覆盖{len(ct.all_diff.exprs)}个变量")
跑出来你就能看到对应9行的9个全不同约束明明白白列在里面,每个约束覆盖的变量也和你代码里传的网格单元格完全对应。
补充个小细节:所有
AddXxx开头的约束添加方法,调用的时候就会立刻解析你传进去的变量表达式,完成约束实体的创建和注册,不会等求解的时候才生成,也不会因为循环里的临时变量销毁就把约束丢了,只要是调用方法成功添加的约束,都能通过上面的接口查到。
内容的提问来源于stack exchange,提问作者user17135505
相关产品推荐
相关产品推荐

