baostock
时间: 2023-08-24 17:08:19 浏览: 79
Baostock 是一个基于 Python 的股票数据接口,可以获取国内外股票、期货等金融数据,并且提供了一些数据可视化和分析的功能。使用 Baostock 前,需要先注册一个账号并获取 AppKey。
以下是使用 Baostock 获取股票历史行情数据并保存为 CSV 文件的示例代码:
```python
import baostock as bs
import pandas as pd
# 登录系统
lg = bs.login()
if lg.error_code != '0':
print('登录失败')
exit()
# 获取沪深A股历史K线数据
rs = bs.query_history_k_data_plus('sh.600000', 'date,open,high,low,close,volume', start_date='2020-01-01', end_date='2021-12-31', frequency='d', adjustflag='3')
if rs.error_code != '0':
print('请求失败')
exit()
# 打印结果集
data_list = []
while (rs.error_code == '0') & rs.next():
row = rs.get_row_data()
data_list.append(row)
df = pd.DataFrame(data_list, columns=rs.fields)
# 保存数据到 CSV 文件
df.to_csv('sh.600000.csv', index=False)
# 登出系统
bs.logout()
```
在上面的示例代码中,我们先使用 `bs.login()` 方法登录 Baostock 系统,然后调用 `bs.query_history_k_data_plus()` 方法获取指定股票的历史行情数据,并将数据存储在 pandas 的 DataFrame 中。最后,我们使用 `df.to_csv()` 方法将数据保存为 CSV 文件。注意,Baostock 的接口返回的数据类型是 `bscore.resultset.ResultSet`,需要通过循环遍历该对象获取数据。最后,我们使用 `bs.logout()` 方法退出 Baostock 系统。