【Python小技巧】使用sqlite的内存数据库提高股票行情数据读写(附代码)
import sqlite3
# 连接到内存中的SQLite数据库
conn = sqlite3.connect(':memory:')
cursor = conn.cursor()
# 创建股票行情数据表
cursor.execute('''
CREATE TABLE stock_data (
symbol text,
time integer,
open real,
high real,
low real,
close real,
volume real,
PRIMARY KEY (symbol, time)
)
''')
# 插入股票行情数据
cursor.execute('''
INSERT INTO stock_data (symbol, time, open, high, low, close, volume)
VALUES (?, ?, ?, ?, ?, ?, ?)
''', ('AAPL', 1610000000, 123.45, 123.65, 123.45, 123.55, 1000000))
# 查询股票行情数据
cursor.execute('''
SELECT * FROM stock_data WHERE symbol = ? AND time = ?
''', ('AAPL', 1610000000))
row = cursor.fetchone()
print(row) # 输出: ('AAPL', 1610000000, 123.45, 123.65, 123.45, 123.55, 1000000)
# 关闭数据库连接
conn.close()
这段代码演示了如何在Python中使用sqlite3库与内存数据库进行交互。首先,我们创建了一个内存数据库连接,并定义了一个股票行情数据表。接着,我们向表中插入了一条股票行情数据。最后,我们通过查询来验证数据是否已经正确插入,并在完成操作后关闭了数据库连接。这个例子非常适合作为学习如何在Python中使用SQLite内存数据库的教学材料。
评论已关闭