Skip to content

Instantly share code, notes, and snippets.

@jdbfsilva
Forked from pnavarro-algometrics/numpy-sqlite3.py
Created October 9, 2020 19:37
Show Gist options
  • Save jdbfsilva/76455faaa6f19929bb93a7cd47a84be6 to your computer and use it in GitHub Desktop.
Save jdbfsilva/76455faaa6f19929bb93a7cd47a84be6 to your computer and use it in GitHub Desktop.
Insert numpy array into a sqlite3 database
import sqlite3
import numpy
# Array of 4 columns and 100 rows
data = numpy.random.rand(100, 4)
# Create a sample database
conn = sqlite3.connect('/tmp/sample.db')
cursor = conn.cursor()
# Create a new table with four columns
cursor.execute('''create table data (field1 real, field2 real, field3 real, field4 real)''')
conn.commit()
# Insert the data array into the 'data' table
cursor.executemany('''insert into data values (?, ?, ?, ?)''', map(tuple, data.tolist()))
conn.commit()
cursor.close()
conn.close()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment