minimize函数约束条件的循环列写(Python)

minimize函数约束条件的循环列写(Python)

我们知道python的scipy.optimize库中的minimize函数经常用来解决最优化问题,它对约束的条件有一定的格式,详见大佬的例子
https://blog.csdn.net/HappyRocking/article/details/92574229?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param
但是当变量很多,但是约束又相同时,我们如何来列写约束条件呢?总不能一条条地输进去吧?之前我是先循环列写字符串格式的约束,再用eval来执行,最近经过老师一句话指点,我学会了利用map来更加简单地列写,分享给大家。
map的功能详见:https://blog.csdn.net/quanlingtu1272/article/details/95482253?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param
简单说就是把一个参数代入函数来执行,当这个参数是list格式时,就可以实现循环执行某个函数并返回一串结果。
具体如何来循环列写约束条件见下面的代码,我举了一个简单的例子:

#约束条件 0<=xi<=20,i=1,...,5 这里一共五个变量,从x1到x5,拥有相同格式的约束条件  
#目标函数 xi求和最大  
from scipy.optimize import minimize  
  
number=[]  
for i in range(5):        #写一个列表来存需要循环的数字,本例是12345  
    number.append(i)  
      
def cons1(a): #列写第一种约束  
    return {'type': 'ineq', 'fun': lambda x: x[a]-0}  
  
def cons2(a): #列写第二种约束  
    return {'type': 'ineq', 'fun': lambda x: 20-x[a]}  
#后面其他约束同理  
  
def function(n,x):  #定义目标函数  
    purpose=0  
    for i in range(n):  
        purpose+=x[i]  #这里定义目标函数为累加  
    return purpose  
  
def zeros(n):  #给定变量初值  
    x0=[]  
    for i in range(n):  
        x0.append(0)  #这里假设给定各变量初值为零  
    return x0  
  
n=5  
fun=lambda x:-function(n,x) #因为是求最大值,而minimize是求最小值,所以加个负号  
x0=zeros(n)  
#利用map将number列表代入约束中,由于number的元素是1,2,3,...,这相当于循环列写了约束  
a=list(map(cons1,number))  #python3不要忘记用list  
b=list(map(cons2,number))  
#有更多约束时同理  
c=a+b #将约束列表合为一个列表  
res = minimize(fun,x0,constraints=c) # 求解最优化问题 这里约束使用了list形式  
print(res) #打印输出结果

输出结果如下:

fun: -99.99999999999979
     jac: array([-1., -1., -1., -1., -1.])
 message: 'Optimization terminated successfully.'
    nfev: 28
     nit: 4
    njev: 4
  status: 0
 success: True
       x: array([20., 20., 20., 20., 20.])

这里各个x均取最大值20,目标函数最大值为100
本人不才,望各位指正,欢迎交流!!!

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值