我想从数据库中的表中获取列名列表。使用编译指示,我会得到一个元组列表,其中包含很多不需要的信息。有没有办法只获取列名?所以我最终可能会遇到这样的事情:
[Column1,Column2,Column3,Column4]
之所以绝对需要此列表,是因为我想在列表中搜索列名并获取索引,因为很多代码中都使用了索引。
有没有办法得到这样的清单?
谢谢
您可以使用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]