python中sqllite插入numpy數組到數據庫的實現方法

更新時間：2021年06月21日 16:44:00 作者：修煉之路

本文給大家介紹python中sqllite插入numpy數組到數據庫的實現方法，在文章底部給大家提到了Python 操作sqlite數據庫及保存查詢numpy類型數據的實例代碼，需要的朋友參考下吧

sqllite里面并沒有與numpy的array類型對應的數據類型，通常我們都需要將數組轉換為text之后再插入到數據庫中，或者以blob類型來存儲數組數據，除此之外我們還有另一種方法，能夠讓我們直接以array來插入和查詢數據，實現代碼如下

import sqlite3
import numpy as np
import io

def adapt_array(arr):
    out = io.BytesIO()
    np.save(out, arr)
    out.seek(0)
    return sqlite3.Binary(out.read())

def convert_array(text):
    out = io.BytesIO(text)
    out.seek(0)
    return np.load(out)


# 當插入數據的時候將array轉換為text插入
sqlite3.register_adapter(np.ndarray, adapt_array)

# 當查詢數據的時候將text轉換為array
sqlite3.register_converter("array", convert_array)


#連接數據庫
con = sqlite3.connect("test.db", detect_types=sqlite3.PARSE_DECLTYPES)
cur = con.cursor()

#創(chuàng)建表
cur.execute("create table test (arr array)")

#插入數據
x = np.arange(12).reshape(2,6)
cur.execute("insert into test (arr) values (?)", (x, ))

#查詢數據
cur.execute("select arr from test")
data = cur.fetchone()[0]

print(data)
# [[ 0  1  2  3  4  5]
#  [ 6  7  8  9 10 11]]
print(type(data))
# <type 'numpy.ndarray'>

實例代碼看下Python 操作sqlite數據庫及保存查詢numpy類型數據

# -*- coding: utf-8 -*-
'''
Created on 2019年3月6日

@author: Administrator
'''
import sqlite3
import numpy as np
import io

def adapt_array(arr):

    out = io.BytesIO()
    np.save(out, arr)
    out.seek(0)
    return sqlite3.Binary(out.read())

def convert_array(text):
    out = io.BytesIO(text)
    out.seek(0)
    return np.load(out)

# 創(chuàng)建數據庫連接對象
conn = sqlite3.connect('sample_database.db', detect_types=sqlite3.PARSE_DECLTYPES)  # 連接到SQLite數據庫
'''
sqlite3.PARSE_DECLTYPES
本常量使用在函數connect()里，設置在關鍵字參數detect_types上面。表示在返回一行值時，是否分析這列值的數據類型定義。如果設置了本參數，就進行分析數據表列的類型，并返回此類型的對象，并不是返回字符串的形式。

sqlite3.PARSE_COLNAMES 
本常量使用在函數connect()里，設置在關鍵字參數detect_types上面。表示在返回一行值時，是否分析這列值的名稱。如果設置了本參數，就進行分析數據表列的名稱，并返回此類型的名稱
'''
# 參數:memory:來創(chuàng)建一個內存數據庫
# conn = sqlite3.connect(":memory:", detect_types=sqlite3.PARSE_DECLTYPES)

# Converts np.array to TEXT when inserting
sqlite3.register_adapter(np.ndarray, adapt_array)

# Converts TEXT to np.array when selecting
sqlite3.register_converter("array", convert_array)

x = np.arange(12).reshape(2, 6)

# conn = sqlite3.connect(":memory:", detect_types=sqlite3.PARSE_DECLTYPES)
cursor = conn.cursor()
# 創(chuàng)建數據庫表
cursor.execute("create table test (arr array)")
# 插入一行數據
cursor.execute("insert into test (arr) values (?)", (x,))
# 提交
conn.commit()

cursor.execute("select arr from test")
data = cursor.fetchone()[0]

print(data)
'''
[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]]
'''
print(type(data))
'''
<class 'numpy.ndarray'>
'''
cursor.close()  # 關閉Cursor
conn.close()  # 關閉數據庫

以上就是python中sqllite插入numpy數組到數據庫的實現方法的詳細內容，更多關于python numpy數組的資料請關注腳本之家其它相關文章！

您可能感興趣的文章: