预取数据

Question

我从presto命令中得到了这样的结果：

  a| b| c  
 --+--+------
 1 | 3| 6 
 2 | 4| 5 

我知道所有数据的cursor.fetchall()和单行的cursor.fetchone()。

现在，我想从一个特定的列中获取所有数据，例如[1, 2]

有办法做到这一点吗？

Answer 1

看到了一些东西并对其进行了修改以解决问题

class reg(object):
 def __init__(self, cursor, row):
    for (attr, val) in zip((d[0] for d in cursor.description), row) :
        setattr(self, attr, val)

并有一个循环来获得列

for row in cursor.fetchall():
  r = reg(cursor, row)
  print r.a