python中sqllite插入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() #创建表 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) # 创建数据库连接对象 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:来创建一个内存数据库 # 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() # 创建数据库表 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数组的资料请关注我们其它相关文章!
赞 (0)