使用Python求解带约束的最优化问题详解
作者:深山里的小白羊 时间:2022-06-28 18:38:19
题目:
1. 利用拉格朗日乘子法
#导入sympy包,用于求导,方程组求解等等
from sympy import *
#设置变量
x1 = symbols("x1")
x2 = symbols("x2")
alpha = symbols("alpha")
beta = symbols("beta")
#构造拉格朗日等式
L = 10 - x1*x1 - x2*x2 + alpha * (x1*x1 - x2) + beta * (x1 + x2)
#求导,构造KKT条件
difyL_x1 = diff(L, x1) #对变量x1求导
difyL_x2 = diff(L, x2) #对变量x2求导
difyL_beta = diff(L, beta) #对乘子beta求导
dualCpt = alpha * (x1 * x1 - x2) #对偶互补条件
#求解KKT等式
aa = solve([difyL_x1, difyL_x2, difyL_beta, dualCpt], [x1, x2, alpha, beta])
#打印结果,还需验证alpha>=0和不等式约束<=0
for i in aa:
if i[2] >= 0:
if (i[0]**2 - i[1]) <= 0:
print(i)
结果:
(-1, 1, 4, 6)
(0, 0, 0, 0)
2. scipy包里面的minimize函数求解
from scipy.optimize import minimize
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import pyplot as plt
#目标函数:
def func(args):
fun = lambda x: 10 - x[0]**2 - x[1]**2
return fun
#约束条件,包括等式约束和不等式约束
def con(args):
cons = ({'type': 'ineq', 'fun': lambda x: x[1]-x[0]**2},
{'type': 'eq', 'fun': lambda x: x[0]+x[1]})
return cons
#画三维模式图
def draw3D():
fig = plt.figure()
ax = Axes3D(fig)
x_arange = np.arange(-5.0, 5.0)
y_arange = np.arange(-5.0, 5.0)
X, Y = np.meshgrid(x_arange, y_arange)
Z1 = 10 - X**2 - Y**2
Z2 = Y - X**2
Z3 = X + Y
plt.xlabel('x')
plt.ylabel('y')
ax.plot_surface(X, Y, Z1, rstride=1, cstride=1, cmap='rainbow')
ax.plot_surface(X, Y, Z2, rstride=1, cstride=1, cmap='rainbow')
ax.plot_surface(X, Y, Z3, rstride=1, cstride=1, cmap='rainbow')
plt.show()
#画等高线图
def drawContour():
x_arange = np.linspace(-3.0, 4.0, 256)
y_arange = np.linspace(-3.0, 4.0, 256)
X, Y = np.meshgrid(x_arange, y_arange)
Z1 = 10 - X**2 - Y**2
Z2 = Y - X**2
Z3 = X + Y
plt.xlabel('x')
plt.ylabel('y')
plt.contourf(X, Y, Z1, 8, alpha=0.75, cmap='rainbow')
plt.contourf(X, Y, Z2, 8, alpha=0.75, cmap='rainbow')
plt.contourf(X, Y, Z3, 8, alpha=0.75, cmap='rainbow')
C1 = plt.contour(X, Y, Z1, 8, colors='black')
C2 = plt.contour(X, Y, Z2, 8, colors='blue')
C3 = plt.contour(X, Y, Z3, 8, colors='red')
plt.clabel(C1, inline=1, fontsize=10)
plt.clabel(C2, inline=1, fontsize=10)
plt.clabel(C3, inline=1, fontsize=10)
plt.show()
if __name__ == "__main__":
args = ()
args1 = ()
cons = con(args1)
x0 = np.array((1.0, 2.0)) #设置初始值,初始值的设置很重要,很容易收敛到另外的极值点中,建议多试几个值
#求解#
res = minimize(func(args), x0, method='SLSQP', constraints=cons)
#####
print(res.fun)
print(res.success)
print(res.x)
# draw3D()
drawContour()
结果:
7.99999990708696
True
[-1.00000002 1.00000002]
来源:https://blog.csdn.net/qq_33757398/article/details/82834888
标签:Python,约束,优化
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
CSS 的优先规则
2009-01-08 12:40:00
PHP设计模式之模板方法模式Template Method Pattern详解
2023-05-25 00:24:26
使用JavaScript实现旋转的彩圈特效
2023-08-22 16:34:53
![](https://img.aspxhome.com/file/2023/9/56129_0s.png)
Thinkphp结合ajaxFileUpload实现ajax异步图片传输全套代码
2023-06-14 04:01:53
你应当了解的5个CSS3新技术
2009-02-11 13:01:00
![](https://img.aspxhome.com/file/UploadPic/20092/11/css31-preview-80.png)
asp源码实现Access数据库的建立或压缩
2007-08-06 16:54:00
Oracle数据库系统使用经验六则
2010-07-26 13:22:00
ASP中3种分页显示的性能比较
2007-08-15 13:37:00
兼容IE和FF的收藏本站、设为首页代码
2009-01-07 14:14:00
交互设计杂七杂八
2010-09-25 18:41:00
![](https://img.aspxhome.com/file/UploadPic/20109/25/02-24s.jpg)
js断点调试心得分享(必看篇)
2023-07-06 22:13:25
![](https://img.aspxhome.com/file/2023/6/55896_0s.png)
go之如何设置GOROOT和GOPATH
2023-07-19 21:27:24
![](https://img.aspxhome.com/file/2023/3/99633_0s.png)
互联网产品设计零碎记
2010-09-25 12:49:00
![](https://img.aspxhome.com/file/UploadPic/20109/25/diedai-600x64-87s.png)
Goland激活码破解永久版及安装详细教程(亲测可以)
2023-06-18 14:30:43
![](https://img.aspxhome.com/file/2023/7/86287_0s.png)
轻松解决Mysql“信息包过大”的处理方法
2008-06-07 16:50:00
Python对两个有序列表进行合并和排序的例子
2022-06-07 00:11:37
Python NumPy实用函数笔记之allclose
2023-08-24 18:09:12
MYSQL 字符串操作
2008-11-13 12:34:00
ThinkPHP模板判断输出Empty标签用法详解
2023-11-14 20:46:54
简化ADO数据库操作的控件(带分页功能)
2008-05-20 13:15:00