python sqlalchemy动态获取列名?

时间:2012-01-20 20:42:47

标签: python sqlalchemy associative

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from sqlalchemy import create_engine

connection = create_engine('mysql://user:passwd@localhost:3306/db').connect()

result = connection.execute("select * from table")
for v in result:
        print v['id']
        print v['name']
connection.close()

我如何动态地获得TABLES COLUMNS NAMES?在这种情况下idname

2 个答案:

答案 0 :(得分:14)

您可以通过调用result.keys()找到列,也可以通过调用v.keys()循环内的for来访问这些列。

以下是使用items()的示例:

for v in result:
    for column, value in v.items():
        print('{0}: {1}'.format(column, value))

答案 1 :(得分:1)

最直接的解决方案

没有比使用列表理解的单线解决方案更简单的解决方案了。这也是最直接的方法:

[col for col in result.keys()]
# return: ['id', 'name']

@Saul的答案也可以,但是您需要谨慎地仅对每个cursor.description中的第一个元素进行迭代,以免在返回列表的每个元组中得到一堆None

它的效率也较低,因为您需要遍历ResultProxy,访问cursor.description属性,并且每个属性仅检索索引为0的元素。

在Python中使用timeit进行了500,000次迭代,显示出速度差异(0.016与0.011):

connection = create_engine('sqlite:///rcsample.db').connect()
result = connection.execute("select * from response")
def cursfunc():
    return [ i[0] for i in result.cursor.description ]
print(timeit.timeit("cursfunc()", setup="from __main__ import cursfunc", number=500000))
# return: 0.01606178

虽然建议的解决方案在约30%的时间内完成:

connection = create_engine('sqlite:///rcsample.db').connect()
result = connection.execute("select * from response")

def keysfunc():
    return [col for col in result.keys()]
print(timeit.timeit("keysfunc()", setup="from __main__ import cursfunc", number=500000))
# return: 0.01097001

实际上,我的怀疑是,具有更多列的表上的时间差可能比上面明显简化的示例更大。

实践中:键和值

实际上,您可能希望动态打印键和值。有两种解决方法。第一个:

results = conn.execute('SELECT * FROM salesperson')
[{column:value for column, value in result.items()} for result in results]
# returns: [{'id': 1, 'name': 'John Doe'}, {'id': 2, 'name': 
# 'Margaret'}, {'id': 3, 'name': 'Anna'}]

或者使用拆包:

rows = conn.execute('SELECT * FROM salesperson LIMIT 2').fetchall()
print([{**row} for row in rows])
# returns: [{'id': 1, 'name': 'John Doe'}, {'id': 2, 'name': 'Margaret'}]

这两种方法都是直接的和pythonic的,同时也使程序员不必显式指定(或事先知道)列名。