我正在使用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']]
但是,我也想要列名。我读了另一个类似的问题here,但在尝试以下我得到一个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
关于如何获取列名的任何想法?
要获取所有列名,可以使用方法keys
作为行项,例如:
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])