baostock加pandas取股票日线数据存入数据库
时间: 2023-06-29 08:04:06 浏览: 168
首先,你需要安装baostock和pandas库,可以使用以下命令进行安装:
```
pip install baostock pandas
```
然后,你需要在baostock官网注册并获取自己的token,用于调用API接口。
接下来,你可以使用以下代码获取股票日线数据,并将其存入MySQL数据库中(需要先安装MySQL数据库及相关库):
```python
import baostock as bs
import pandas as pd
from sqlalchemy import create_engine
# 登录baostock系统
lg = bs.login()
# 设置要获取的股票代码和起始日期(这里以000001.SZ和2019年1月1日为例)
code = '000001.SZ'
start_date = '2019-01-01'
# 调用baostock API获取股票日线数据
rs = bs.query_history_k_data_plus(code, "date,open,high,low,close,volume", start_date=start_date, frequency="d", adjustflag="3")
data_list = []
while (rs.error_code == '0') & rs.next():
# 将获取到的数据存入列表中
data_list.append(rs.get_row_data())
# 将数据列表转换为DataFrame格式
df = pd.DataFrame(data_list, columns=rs.fields)
# 将数据存入MySQL数据库中
engine = create_engine('mysql+pymysql://username:password@localhost:3306/dbname')
df.to_sql('stock_daily_data', engine, if_exists='append', index=False)
# 登出baostock系统
bs.logout()
```
需要注意的是,上述代码中的`username`、`password`、`localhost`和`dbname`需要根据你自己的情况进行修改。
阅读全文