Python将numpy数组插入sqlite3数据库

小编典典

Python将numpy数组插入sqlite3数据库

python

我试图在sqlite3数据库中存储大约1000个浮点数的numpy数组，但我不断收到错误“
InterfaceError：错误绑定参数1-可能是不受支持的类型”。

我给人的印象是BLOB数据类型可以是任何东西，但绝对不适用于numpy数组。这是我尝试过的：

import sqlite3 as sql
import numpy as np
con = sql.connect('test.bd',isolation_level=None)
cur = con.cursor()
cur.execute("CREATE TABLE foobar (id INTEGER PRIMARY KEY, array BLOB)")
cur.execute("INSERT INTO foobar VALUES (?,?)", (None,np.arange(0,500,0.5)))
con.commit()

我是否可以使用另一个模块将numpy数组放入表中？还是可以将numpy数组转换为sqlite可以接受的Python中的另一种形式（例如可以拆分的列表或字符串）？性能不是重点。我只是想让它工作！

谢谢！

阅读 216

2021-01-20

共1个答案

小编典典

您可以使用来注册新的array数据类型sqlite3：

import sqlite3
import numpy as np
import io

def adapt_array(arr):
    """
    http://stackoverflow.com/a/31312102/190597 (SoulNibbler)
    """
    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)


# 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)

con = sqlite3.connect(":memory:", detect_types=sqlite3.PARSE_DECLTYPES)
cur = con.cursor()
cur.execute("create table test (arr array)")

使用此设置，您可以简单地插入NumPy数组，而无需更改语法：

cur.execute("insert into test (arr) values (?)", (x, ))

并直接从sqlite作为NumPy数组检索数组：

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'>

2021-01-20