python - 从 Vertica 数据库获取列名(标题)？

Question

在通过 sql 查询从 python 中的 vertica 数据库中提取数据时，我试图提取列名。我正在使用 vertica-python 0.6.8。到目前为止，我正在创建第一行的字典，但我想知道是否有更简单的方法。这就是我现在的做法:

import vertica_python
import csv
import sys
import ssl
import psycopg2

conn_info = {'host': '****',
             'port': 5433,
             'user': '****',
             'password': '****',
             'database': '****',
             # 10 minutes timeout on queries
             'read_timeout': 600,
             # default throw error on invalid UTF-8 results
             'unicode_error': 'strict',
             # SSL is disabled by default
             'ssl': False}

connection = vertica_python.connect(**conn_info)
cur = connection.cursor('dict')
str = "SELECT * FROM something WHERE something_happens LIMIT 1"
cur.execute(str)
temp = cur.fetchall()
ColumnList = []
for column in temp[0]:
    ColumnList.append(column)

干杯

Answer 1

两种方式:

首先，如果您想要列列表，您可以只访问字典的键，这基本上就像您拥有的一样，但更短:

ColumnList = temp[0].keys()

其次，您可以访问游标的字段列表，我认为这才是您真正要找的:

ColumnList = [d.name for d in cur.description]

第二个更好，因为即使结果为空，它也能让您看到列。