Python/Sqlite3 SELECT時のカラム名（ヘッダー）の表示方法について

Question

前提

pythonでsqliteのDBからデータをSELECTする際に
カラム名を表示したいのですが表示させることができず。
お知恵をお貸しください。

実現したいこと

該当ソースコードにて以下のように
nameやageなどのカラム名を表示させたい。

name age
yamada 20

該当のソースコード

import os
import sqlite3

db_name = 'test.db'
current_dir = os.path.dirname(__file__)
dbpath = os.path.join(current_dir, db_name)
conn = sqlite3.connect(dbpath)
conn.row_factory = sqlite3.Row
cur = conn.cursor()

cur.execute('DROP TABLE IF EXISTS persons')
cur.execute('CREATE TABLE IF NOT EXISTS persons(name, age)')
cur.execute('INSERT INTO persons VALUES("yamada", "20")')

cur.execute("SELECT * FROM persons")

for row in cur.fetchall():
    print('\t'.join(row))

conn.commit()
cur.close()
conn.close()

# result ->
# yamada  20

試したこと

sqliteの機能で.headers onで表示されるとのことだったので

.headers on
conn.headers on
cur.headers on

などSELECT前後に追加してみましたが解決せず。（SyntaxError）

Answer 1

cursor.descriptionに列名が含まれているので、以下のように出力できます。

cols = [d[0] for d in cur.description]
print('\t'.join(cols))
for row in cur.fetchall():
    print('\t'.join(row))
#name    age
#yamada  20

あるいはpandasを利用するとdf = pd.read_sql('SELECT * FROM persons', conn)で列名含まれたデータフレームが得られるので楽かもしれません。

Comments

[sikimori]

早速の回答ありがとうございます！
調べていてdescriptionをforで回すのはどうもスマートじゃないよなあと思っていた所でした。
pandasを使ってみます！