sqllite里面并没有与numpy的array类型对应的数据类型，通常我们都需要将数组转换为text之后再插入到数据库中，或者以blob类型来存储数组数据，除此之外我们还有另一种方法，能够让我们直接以array来插入和查询数据，实现代码如下
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
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类型数据
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
# -*- 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数组的资料请关注服务器之家其它相关文章！
原文链接：https://blog.csdn.net/sinat_29957455/article/details/118027857
