I am working on a LP model with pyomo,but when I create constraint, it shows a key error of 'can not find a certain combination'. I know list all the combinations can solve this problem. But the real data has many combinations. Is there any easy way to deal with this knid of problem? thanks!Here is a simple example:
This is a similar question for reference: Is there an easier way to avoid this kind of index key error of pyomo?
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)
#constraint2
def cons2(model,a):
return(sum(model.AB[a,b]<=C[a,b] for b in B)<=1)
model.Cons2 = Constraint(A,rule = cons2)