AMPL与Python - 导入表(多维词典?)

时间:2013-07-20 16:22:55

标签: python set mathematical-optimization ampl gurobi

我是AMPL用户,尝试使用Python(我的第一个Python代码)编写线性编程优化模型。我试图找到如何在复合集上声明索引参数。例如,在AMPL中,我会说:     设置A.     设置B.     设置C.     param x {A,B,C}     param y {A,B,C}     param z {A,B,C} 可以通过AMPL从数据库轻松读取上述设置和参数。

我从数据库中读取的表有六个字段,即A,B,C,x,y,z。其中三个是主键(A,B,C),其余(x,y,z)是在主键上索引的值。

PYTHON PART: 我正在使用PYODBC模块连接SQL Server。我试过“dict”但它只能索引一个键。 我不确定应该使用哪个python功能将前三个字段声明为复合集,将x,y和z声明为在复合集上索引的值。

import pyodbc    
con = pyodbc.connect('Trusted_Connection=yes', driver = '{SQL Server Native Client 10.0}', server = 'Server', database='db')
cur = con.cursor()
cur.execute("execute dbo.SP @Param =%d" %Param)
result = cur.fetchall()
Comp_Key, x, y, z= dict((A, B, C, [x,y,z]) for A, B, C, x, y, z in result)

当然这是不正确的。我想不出办法做到这一点。

请帮帮我:) 提前致谢!

1 个答案:

答案 0 :(得分:2)

取代这个:

Comp_Key, x, y, z= dict((A, B, C, [x,y,z]) for A, B, C, x, y, z in result)

你可以这样做(如果A,B,C!= B,A,C,即A B C的顺序很重要):

final_result = dict(((A, B, C), [x, y, z]) for A, B, C, x, y, z in result)

OR

final_result = {(A, B, C): [x, y, z] for A, B, C, x, y, z in result}  # more readable than first one

或者你可以这样做(如果A,B,C == B,A,C,即A B C的顺序无关紧要):

final_result = dict((frozenset(A, B, C), [x, y, z]) for A, B, C, x, y, z in result)

OR

final_result = {frozenset(A, B, C): [x, y, z] for A, B, C, x, y, z in result}  # more readable than first one