pymysql 保存股票数据

pymysql是一个Python连接和操作MySQL数据库的库。若要使用pymysql保存股票数据，首先需要在Python环境中安装pymysql库。

安装好pymysql后，可以使用以下步骤来保存股票数据：

1. 导入必要的库和模块：

import pymysql

2. 连接到MySQL数据库：

db = pymysql.connect(host='localhost', user='root', password='123456', database='stock_data')

需要提供MySQL数据库的主机名、用户名、密码以及要连接的数据库名称。

3. 创建一个游标对象：

cursor = db.cursor()

4. 创建一个表来存储股票数据：

sql = '''CREATE TABLE IF NOT EXISTS stocks (
         id INT PRIMARY KEY AUTO_INCREMENT,
         symbol VARCHAR(10) NOT NULL,
         date DATE NOT NULL,
         close FLOAT NOT NULL,
         volume INT NOT NULL
         )'''
cursor.execute(sql)

在此示例中，我们创建了一个名为stocks的表，它包含股票代码(symbol)、日期(date)、收盘价(close)和交易量(volume)的列。

5. 插入股票数据：

data = [
    ('AAPL', '2021-01-01', 135.0, 1000000),
    ('AAPL', '2021-01-02', 136.0, 1500000),
    ('AAPL', '2021-01-03', 134.0, 1200000)
]
sql = "INSERT INTO stocks (symbol, date, close, volume) VALUES (%s, %s, %s, %s)"

try:
    cursor.executemany(sql, data)
    db.commit()
    print("数据插入成功")
except:
    db.rollback()
    print("数据插入失败")

这里我们需要提供股票数据的元组列表，每个元组包含股票代码、日期、收盘价和交易量。

6. 关闭数据库连接：

db.close()

以上是使用pymysql保存股票数据的基本步骤。根据具体需求，我们可以进一步优化代码，增加异常处理、查询和更新等操作。