快速获取数据

presto fetch data

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

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

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

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

有办法吗?

看到东西修改一下解决了问题

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