python - 有没有办法在 sqlite 中获取列名列表？

Question

我想从数据库中的表中获取列名列表。使用 pragma 我得到一个包含大量不需要信息的元组列表。有没有办法只获取列名？所以我可能会得到这样的结果:

[Column1, Column2, Column3, Column4]

之所以我绝对需要这个列表，是因为我想在列表中搜索一个列名并获取索引，因为我的很多代码中都使用了索引。

有没有办法得到这样的列表？

谢谢

Answer 1

您可以使用 sqlite3 和 pep-249

import sqlite3
connection = sqlite3.connect('~/foo.sqlite')
cursor = connection.execute('select * from bar')

cursor.description 是列的描述

names = list(map(lambda x: x[0], cursor.description))

您也可以使用列表推导:

names = [description[0] for description in cursor.description]