環境

Python3.6
SQLite3

やりたいこと

作成したテーブルのカラムタイトル（言葉は違うかも・・・）を知りたい。

python
1import sqlite3
2dbpath='database.db'
3con = sqlite3.connect(dbpath)
4cursor = con.cursor()
5#新しくテーブルを作る
6cursor.execute('CREATE TABLE sample (hoge,huga)')
7#でーたをブッコム
8cursor.execute("INSERT INTO sample VALUES ('hoogee','huugaa')")
9con.commit()
10con.close()

ここで、テーブル名sampleは覚えているのだけれども、hoge,hugaの要素が大量に増えてきた場合に、あれ、なんだったっけとなってしまうので、
こいつらを知る方法があれば、ご教授いただけると助かります。

よろしくお願いいたします。

行動規範の内容に同意します

回答1件

ベストアンサー

データベースを扱うときに、カラム名が分からない状態でプログラムを書くことはあまりないと思いますが、以下のように書くと列名と取得した値を関連づけたデータにできます。

参考: https://docs.python.jp/3/library/sqlite3.html#sqlite3.Cursor.description

Python
1>>> cursor = con.cursor()
2>>> cursor.execute('SELECT * FROM sample')
3<sqlite3.Cursor object at 0x000001F8F3C45730>
4>>> row = cursor.fetchone()
5>>> row
6('hoogee', 'huugaa')
7>>> cursor.description
8(('hoge', None, None, None, None, None, None), ('huga', None, None, None, None, None, None))

Python
1>>> cursor.execute("INSERT INTO sample VALUES ('hello','world')")
2<sqlite3.Cursor object at 0x000001F8F3C45490>
3>>> results = []
4>>> cursor.execute('SELECT * FROM sample')
5<sqlite3.Cursor object at 0x000001F8F3C45490>
6>>> for row in cursor.fetchall():
7...     x = dict(zip([d[0] for d in cursor.description], row))
8...     results.append(x)
9...
10>>> results
11[{'hoge': 'hoogee', 'huga': 'huugaa'}, {'hoge': 'hello', 'huga': 'world'}]

row_factoryを使うと以下のように簡潔に書けます。

参考: https://docs.python.jp/3/library/sqlite3.html#row-objects

Python
1>>> con.row_factory = sqlite3.Row
2>>> cursor = con.cursor()
3>>> cursor.execute('SELECT * FROM sample')
4<sqlite3.Cursor object at 0x000001F8F3C45650>
5>>> results = []
6>>> for row in cursor.fetchall():
7...     results.append(dict(x))
8...
9>>> results
10[{'hoge': 'hello', 'huga': 'world'}, {'hoge': 'hello', 'huga': 'world'}]