我正在使用PYOMO进行LP模型,但是当我创建约束时,它显示了'找不到某个组合的关键错误。我知道所有组合都可以解决此问题。但是真实的数据具有许多组合。有什么简单的方法可以处理这种问题吗?谢谢!这是一个简单的例子:
from pyomo.environ import *
import pandas as pd
data = [['tom','A', 10], ['nick','A', 15], ['juli','B',14]]
df = pd.DataFrame(data, columns = ['Name','Type', 'Age'])
#set
A = set(df['Name'])
B = set(df['Type'])
model = ConcreteModel()
#parameter
C= df.set_index(['Name','Type'])['Age'].to_dict()
#varibale
model.AB = Var(A,B,domain = NonNegativeReals)
#constraint1
def cons1(model,a,b):
return(model.AB[a,b]<=C[a,b])
model.Cons1 = Constraint(A,B,rule = cons1)
使用C
字典中的键来定义您的索引集:
from pyomo.environ import *
import pandas as pd
data = [['tom','A', 10], ['nick','A', 15], ['juli','B',14]]
df = pd.DataFrame(data, columns = ['Name','Type', 'Age'])
model = ConcreteModel()
#parameter
C= df.set_index(['Name','Type'])['Age'].to_dict()
#varibale
model.IJ = Set(initialize=C.keys())
model.AB = Var(model.IJ,domain = NonNegativeReals)
#constraint1
def cons1(model,a,b):
return(model.AB[a,b]<=C[a,b])
model.Cons1 = Constraint(model.IJ,rule = cons1)