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 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类型数据
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 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数组的资料请关注服务器之家其它相关文章!
原文链接:https://blog.csdn.net/sinat_29957455/article/details/118027857