Question

我正在尝试从表中获取列

 query = session.prepare("""  SELECT * FROM mytable   """)

 row_data = session.execute(query, )

我想要的是从row_data获取列名。有没有办法做到这一点？。

Answer 1

在大多数python数据库适配器中，您可以使用DictCursor使用类似于Python词典而不是元组的接口来检索记录。

使用cassandra ：

>>> from cassandra.query import dict_factory
>>> session = cluster.connect('mykeyspace')
>>> session.row_factory = dict_factory
>>> rows = session.execute("SELECT name, age FROM users LIMIT 1")
>>> print rows[0]
{u'age': 42, u'name': u'Bob'}

使用psycopg2 ：

>>> dict_cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
>>> dict_cur.execute("INSERT INTO test (num, data) VALUES(%s, %s)",
...                  (100, "abc'def"))
>>> dict_cur.execute("SELECT * FROM test")
>>> rec = dict_cur.fetchone()
>>> rec['id']
1
>>> rec['num']
100
>>> rec['data']
"abc'def"

使用MySQLdb ：

>>> import MySQLdb 
>>> import MySQLdb.cursors 
>>> myDb = MySQLdb.connect(user='andy47', passwd='password', db='db_name', cursorclass=MySQLdb.cursors.DictCursor) 
>>> myCurs = myDb.cursor() 
>>> myCurs.execute("SELECT columna, columnb FROM tablea") 
>>> firstRow = myCurs.fetchone() 
{'columna':'first value', 'columnb':'second value'}

Answer 2

你可以这样，

fields = [ix[0] for ix in cursor.description]

Documentation

cursor.description＃最初没有，表示N个元组的列表执行后连续的N列。只要包含类型和名称信息，而不是值。

Answer 3

使用cassandra-driver 3.14，您还可以通过访问查询结果条目的_fields参数来获取列名。

from cassandra.cluster import Cluster

session = Cluster().connect('mykeyspace')

rows = session.execute('select * from mytable limit 1')

column_names = rows.one()._fields

从python中的表的数据集中获取列名

3 个答案: