PythonでSQLite３のデータを取得し、辞書型に整える方法を記載します。
辞書型は{ key : value }の構成にすることで情報の紐付けができるため、
どのような値なのかが分かりやすくなります。
今回はKeyをDBのカラム名、Valueは格納されているデータの値になります。
目指すゴール
以下のようなデータを作成します。
[
{ "id" : 1, "fruit" : "apple", "price" : 100 },
{ "id" : 2, "fruit" : "banana", "price" : 150 },
{ "id" : 3, "fruit" : "peach", "price" : 120 }
]
for文で扱いやすいようリスト型の中に辞書型が複数存在するデータを作成します。
データの準備
データを作成するスクリプトは以下になります。
insert.py
import sqlite3
db_name = 'test.db'
conn = sqlite3.connect(db_name)
cur = conn.cursor()
cur.execute('CREATE TABLE items(id INTEGER PRIMARY KEY AUTOINCREMENT, fruit STRING, price INTEGER)')
cur.execute('insert into items(fruit, price) values("apple", 100)')
cur.execute('insert into items(fruit, price) values("banana", 150)')
cur.execute('insert into items(fruit, price) values("peach", 120)')
conn.commit()
conn.close()
1つのデータを辞書型にする
いきなり複数データを処理するよりまず1つのデータで、
うまくいくか確認してからfor文で対応するようにします。
以下対応をします。
- DBデータをselect文で取得する
- selectしてきたデータをリスト型に変換する : fetchall関数
- 2つのリストを辞書型に変換 : dict( zip( list1, list2 ) )
import sqlite3
dbname = 'test.db'
conn = sqlite3.connect(dbname)
cur = conn.cursor()
db_col = [ "id", "fruit", "price" ]
cur.execute('select * from items')
select_items = cur.fetchall()
result = dict(zip(db_col, select_items[0]))
print(result)
conn.close()
cur.execute('select * from items')でitemsテーブル内全データを取得します。
sqlite3の関数でfetchall関数で、select文を取得してデータをリスト型をして取得します。
dict(zip(db_col, select_items[0]))でdb_colのリストとfetcall()でリスト化したデータの１番目を辞書型に変換します。
以下が実行結果になります。
{'id': 1, 'fruit': 'apple', 'price': 100}
これで1つのデータを辞書型にできたので、これをfor文で複数データを辞書型にしていきます。
複数データを辞書型にする
for文で辞書型に変換し、リストに追加していきます。
import sqlite3
dbname = 'test.db'
conn = sqlite3.connect(dbname)
cur = conn.cursor()
db_col = [ "id", "fruit", "price" ]
cur.execute('select * from items')
select_items = cur.fetchall()
result_list = [] # 空のリストを作成
for item in select_items:
d_item = dict(zip(db_col, item))
result_list.append(d_item)
print(result_list)
conn.close()
select文で取得したデータをfor文で1つずつ辞書型に変換し、
result_list.appendで作成した空のリストに追加していきます。
以下実行結果
[
{'id': 1, 'fruit': 'apple','price': 100},
{'id': 2, 'fruit': 'banana', 'price': 150},
{'id': 3, 'fruit': 'peach', 'price': 120}
]
参考サイト
https://qiita.com/drafts/8191896ea8577a0213aa/edit
https://note.nkmk.me/python-zip-usage-for/