时间戳

首页 > 解决方案 > psycopg2.extras.DictCursor 没有给我列名

python - psycopg2.extras.DictCursor 没有给我列名

问题描述

我正在使用 psycopg2 访问 Postgres 数据库中的数据。我正在使用 psycopg2.extras.DictCursor 使用以下查询以类似 dict 的形式获取数据：

try:
    self.con = psycopg2.connect(dbname=self.db, user=self.username, 
host=self.host, port=self.port)

cur = self.con.cursor(cursor_factory=psycopg2.extras.DictCursor)

cur.execute("SELECT * FROM card")

result = cur.fetchall()

print result

我得到一个列表的输出：

[['C01', 'card#1', 'description#1', '1'], ['C02', 'card#2', 'description#2', '1']]

但是，我也想要列名。我在这里阅读了另一个类似的问题，但是在尝试以下操作时，我得到了KeyError：

cur.execute("SELECT * FROM card")
for row in cur:
    print(row['column_name'])

但是，以下工作：

cur.execute("SELECT * FROM card")
for row in cur:
    print(row['id'])

输出：

C01
C02

关于如何获取列名的任何想法？

标签： pythonpsycopg2

解决方案

要获取所有列名，您可以将 mehtodkeys用于行项，例如：

cur.execute("SELECT * FROM card")
result = cur.fetchall()
keys = list(result[0].keys()) if result else []
for row in result:
    for key in keys:
        print(row[key])

推荐阅读