【Python】 gurobipy 学习笔记4——prod函数

prod(coeff,pattern)函数的使用

本节我们来介绍prod函数：

prod函数可以起到做到带不同系数的变量求和的目的
coeff为一个dict类型，指定待计算的元素的系数,
coeff的key要与待计算的集合中的key能对应。

使用举例：

#创建模型
m=Model()
#创建变量
xij=m.addVars(2,2,name="Xij") #xij下标为(0,0)(0,1)(1,0)(1,1)
#定义确定每个xij系数的dict
coeff={(0,0):1,
       (0,1):2,
       (1,0):3,
       (1,1):4}
coeff_tupledict=tupledict(coeff)

prod_practice1=xij.prod(coeff_tupledict)
prod_practice2=xij.prod(coeff_tupledict,0,"*")

m.addConstr(prod_practice1<=10)
m.addConstr(prod_practice1>=5)

m.update()
m.write("prod_practice.lp")
print(prod_practice1)
print(prod_practice2)

输出：
<gurobi.LinExpr: Xij[0,0] + 2.0 Xij[0,1] + 3.0 Xij[1,0] + 4.0 Xij[1,1]>
<gurobi.LinExpr: Xij[0,0] + 2.0 Xij[0,1]>

通过prod函数，就可以快速编写变量*系数的公式，同时注意，为了减少出错，应该把系数的dict转化为tupledict。

和sum函数的对比：

#与sum函数的对比
m=Model()
t=[(1,1),(1,2),(1,3),
   (2,1),(2,2),(2,3),
   (3,1),(3,2),(3,3)]
#创建变量
x=m.addVars(t,vtype=GRB.BINARY,name="x")
m.update()
#使用sum函数练习
print("x.sum(1,'*''):",x.sum(1,"*"))

#输出：
x.sum(1,'*''): <gurobi.LinExpr: x[1,1] + x[1,2] + x[1,3]>

总结一下，要想使用prod函数，需要先设定一个索引+对应系数的tupledict，然后根据需要使用函数即可，下一节将会介绍进阶经典项目-qiucksum快速求和。

本系列往期内容:

【Python】 gurobipy 学习笔记1——建模基础_胡琪琪爱编程的博客-CSDN博客

【Python】 gurobipy 学习笔记2——sum的基础使用_胡琪琪爱编程的博客-CSDN博客

【Python】 gurobipy 学习笔记3——sum的进阶使用_胡琪琪爱编程的博客-CSDN博客