Z3求解器结果输出(python)
Z3求解器教程可以参考这里
简单复述一下使用Z3求解器解题的步骤:
- 安装和导入
# 安装
pip install z3-solver
# 导入
from z3 import *
- 声明变量
# 变量为整数
## 逐一定义
a = Int('a')
b = Int('b')
## 一起定义
a,b = Ints('a b')
# 变量为实数
## 逐一定义
c = Real('c')
d= Real('d')
## 一起定义
c,d = Reals('c d')
- 建立模型约束
# 建立求解器
s = Solver()
## 添加约束
s.add(a+d==8)
s.add(a-d==2)
- 模型求解
# 检查模型是否有解
s.check() # 若有解,输出sat;否则输出unsat
Out[14]: sat
# 结果计算
s.model()
Out[15]: [a = 5, d = 3]
可以看到求解器计算出了正确结果,但是变量的类型是z3.ArithRef,python无法将其作为变量继续调用。
以下转换可将结果输出为int或fraction类型
m = s.model()
# 输出为int类型
a_int = m[a].as_long()
# 输出为fraction类型
d_float = m[d].as_fraction()
fraction类型的数据可以直接转换成float类型
d_float
Out[30]: Fraction(3, 1)
float(d_float)
Out[31]: 3.0
如果变量解得的值本身就是小数形式的,就不需要float(d_float)这一步了,数据格式直接就是float了
补充
有时候在使用.as_long() / .as_fraction()时会报错,如下例:
In [48]:m[ks]
Out[48]: 1.5590329177?
In [49]:m[ks].as_fraction()
Traceback (most recent call last):
File "<ipython-input-49-1d91f5412d29>", line 1, in <module>
m[ks].as_fraction()
AttributeError: 'AlgebraicNumRef' object has no attribute 'as_fraction'
我目前还不知道是什么原因(猜测可能是结果无法输出精确值?)。
这里提供一个报错后的处理办法:
- 先用.approx()属性对结果取近似,括号中填入保留的小数位数。这样就把数据近似转换成了分式的形式,此时数据类型还是z3.RatNumRef;
aaa = m[ks].approx(20)
aaa
Out[51]: 1884755147893967473528193/1208925819614629174706176
- 再用.as_fraction()就可以把数据转成Fraction类型了;
aaa = aaa.as_fraction()
- 最后float()一下,出来的就是python可用的变量了
这是我目前的解决办法,如果以后发现更好的方法我再来补充。