python中sqllite插入numpy数组到数据库的实现方法

sqllite里面并没有与numpy的array类型对应的数据类型，通常我们都需要将数组转换为text之后再插入到数据库中，或者以blob类型来存储数组数据，除此之外我们还有另一种方法，能够让我们直接以array来插入和查询数据，实现代码如下

import sqlite3import numpy as npimport iodef 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转换为arraysqlite3.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 sqlite3import numpy as npimport iodef 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 insertingsqlite3.register_adapter(np.ndarray, adapt_array)# Converts TEXT to np.array when selectingsqlite3.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() # 关闭Cursorconn.close() # 关闭数据库

以上就是python中sqllite插入numpy数组到数据库的实现方法的详细内容，更多关于python numpy数组的资料请关注好吧啦网其它相关文章！