zoukankan      html  css  js  c++  java
  • 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()  # 关闭数据库
    
    
  • 相关阅读:
    关于TCP中对于ACK报文是否需要确认的理解
    定时器
    几个错误
    C++继承与组合
    Ubuntu18的Redis: 4.0安装
    Ubuntu18.04:MySQL: 5.7安装与卸载并配置远程登录(服务器装Mysql5.7一条龙服务)
    Ubuntu18服务器修改字体大小
    Ubuntu修改系统时间
    Ubuntu18.04防火墙安装关闭开启操作等
    Ubuntu阿里云镜像源配置
  • 原文地址:https://www.cnblogs.com/gmhappy/p/11863987.html
Copyright © 2011-2022 走看看