Get best of the week

Inserir array numpy no banco de dados MySQL via Python

Se você encontrar um problema que normalmente não pode salvar uma matriz numpy em um banco de dados MySQL , então este post é para você! A postagem original foi postada no meu blog .

Eu escolhi um método de salvar para mim através do módulo pickle . Com ele, você pode salvar com segurança uma matriz numpy de qualquer dimensão no blob do banco de dados MySQL.

Então, um exemplo de código:

import mysql.connector
import pickle
import numpy as np

#    
connection = mysql.connector.connect(host='localhost',
                                         database='database',
                                         user='root',
                                         password='')
# 
cursor = connection.cursor()

#       blob
query = 'create table test(`column` blob);'
cursor.execute(query)

#  numpy 
array = np.array([[1,2,3],[4,5,6]])

#    ,   pickle.dumps
query = 'insert into test values(%s);'
db_array = pickle.dumps(array)
cursor.execute(query, [db_array ])

#     test
query = 'select * from test;'
cursor.execute(query)
res = cursor.fetchall()

#  
res

##   : [(bytearray(b'\x80\x03cnumpy.core.multiarray\n_reconstruct\nq\x00cnumpy\nndarray\nq\x01K\x00\x85q\x02C\x01bq\x03\x87q\x04Rq\x05(K\x01K\x02K\x03\x86q\x06cnumpy\ndtype\nq\x07X\x02\x00\x00\x00i8q\x08K\x00K\x01\x87q\tRq\n(K\x03X\x01\x00\x00\x00<q\x0bNNNJ\xff\xff\xff\xffJ\xff\xff\xff\xffK\x00tq\x0cb\x89C0\x01\x00\x00\x00\x00\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x04\x00\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00\x00\x00\x00\x00q\rtq\x0eb.'),)]

#  pickle.load      numpy
pickle.loads(res[0][0])

##  
array([[1, 2, 3],
       [4, 5, 6]])

#                   
cursor.close()
connection.close()

Dessa maneira, você pode trocar matrizes numpy multidimensionais (e regulares) entre Python e MySQL de uma maneira absolutamente desimpedida. Se você ler a imagem usando cv2.imread da biblioteca OpenCV, por exemplo, o princípio permanecerá o mesmo - porque, de fato, é uma matriz multidimensional.

Espero que isso ajude alguém a resolver um problema semelhante!

More articles:


All Articles