通过比较列名来迭代结果集 - 在python中使用Mysqldb

时间:2016-07-19 11:46:55

标签: python iteration mysql-python

我想用python来填充数据库。我必须从表中获取值,计算分数并将此分数填入anoter表。我无法弄清楚如何比较结果集中行的列名。因为我不需要所有的列来计算,但是需要一些其他的id和type_name来进行计算。所以基本上我想这样做:

cur = connection.cursor()
QUERY ="SELECT * FROM table"
cur.execute(QUERY)
rs = cur.fetchall()
for row in rs:
    for col in row:
        // if(col = "X" or col = "Y" or col = "Z"):
               calc ...
        // else:
               use id, type_name whatever ...

如何实现这样的目标?否则,代码会像炸弹一样爆炸。

1 个答案:

答案 0 :(得分:0)

也许有人正在寻找答案。借助之前的评论,我可以像那样解决它

field_names = cur.description 
for row in rs:
    for index, col in enumerate(row):
        name = field_names[index][0]
        if(name == "..."):
        ...
        elif(name == "..."):
        ...