Python 利用Baostock 下载股票代码，写入Mysql数据库

郑连辉241

于 2022-08-06 21:59:38 发布

阅读量4.1k

点赞数 5

分类专栏：学习笔记文章标签：数据库 python mysql

本文链接：https://blog.csdn.net/weixin_71142140/article/details/126200826

版权

本文介绍了如何使用Python的Baostock库下载股票代码，并将其存储到MySQL数据库中。首先通过pip安装baostock，然后导入并使用其query_all_stock函数获取股票代码，将数据转化为DataFrame。接着，利用pymysql连接到已建立的MySQL数据库，创建表并定义函数将股票代码写入数据库。

摘要由CSDN通过智能技术生成

Baostock（证券宝）是一个免费、开源的证券数据平台（无需注册）。详见点这里。

首先，用pip安装baostock。win+r输入cmd进入命令提示符界面。输入pip install baostock

pip install baostock

导入baostock，并且重命名为bs

import baostock as bs

baostock下载股票代码，使用query_all_stock(day=None)函数

data = bs.query_all_stock().get_data()

.get_data()方法把数据直接转化成pandas的DataFrame

定义函数get_stock_codes,该函数返回股票代码列表，该列表可以用来下载股票历史数据

def get_stock_codes(day=None):
    """
    利用baostock获取沪深两市的股票代码
    因为钱不够，所以剔除掉中小板，创业板和北交所的股票代码，指数代码也剔除掉吧
    :day: 日期，默认为空
    :return: 筛选后的股票代码列表
    """

    # 登入baostock
    bs.login()

    # 获取股票代码
    codes = bs.query_all_stock(day=day).get_data()

    # 股票代码筛选，剔除中小板，创业板，北交所以及指数代码
    df = codes[((codes['code'] < 'sz.300000') & (codes['code'] > 'sh689000')) | (
                (codes['code'] < 'sh.688000') & (c