將數據從sqlite導出到numpy數組

Question

我不是程序員，我純粹是出於業余愛好。

我找到了一種將numpy數組保存到sqlite數據庫的方法

import sqlite3
import numpy

# Array of 4 columns and 100 rows
data = numpy.random.rand(100, 4)

# Create a sample database
conn = sqlite3.connect('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()

但是我有一個關於尋找一種方法來逆轉過程的問題。.我想從數據庫中以numpy數組加載數據。.一些建議在哪里找到一個簡單的示例？

Answer 1

只需獲取所有值。 那給你一個元組列表。 np.array()帶您回到原始數組：

In [12]: cursor.execute('SELECT * from data')
Out[12]: <sqlite3.Cursor at 0xaf8e9d60>
In [13]: alist = cursor.fetchall()
In [14]: len(alist)
Out[14]: 100
In [15]: alist[0]
Out[15]: 
(0.3327498114993416,
 0.6164620040846208,
 0.5099007559772143,
 0.7808234554641948)
In [16]: data1 = np.array(alist)
In [17]: np.allclose(data, data1)
Out[17]: True

它是元組列表的事實並不重要。 和列表列表一樣好。