通过python获取Tushare平台的数据

原创

已于 2024-06-12 19:26:53 修改 · 1.8k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#python

于 2021-05-27 10:47:58 首次发布

本文介绍了如何使用Python的tushare库获取证券数据，并将其存储到MySQL数据库中。首先设置tushare API，然后建立数据库连接和数据表，接着将数据转换为字典格式，最后通过循环将数据批量插入数据库。文中还提供了处理数据缺失和更新数据的方法。

早前了解了tushare网站，网站提供证券相关的行业信息及每日交易数据，是金融从业尤其是量化从业人员常用数据网站。尝试从网站上取数搭建自己的金融数据体系，于是有了用从tushare上取数并插入mysql数据库中的尝试，简单记录一下
1、导包

####后续需要用到的库，期间pymysql总是连不上，pip install/uninstall mysql 尝试了多遍无果，最后还是找了个package文件放到目录下才搞定
import tushare as ts
import pandas as pd
import pymysql
ts.set_token('**************************') 
pro = ts.pro_api()

2、连接数据库

####连接数据库，如history已存在就删掉
db = pymysql.connect(host="localhost", user="root",password="123456", database="data", charset='utf8' )
cursor = db.cursor()
cursor.execute("DROP TABLE IF EXISTS history")

3、建表

# 创建数据表SQL语句
sql = """CREATE TABLE history (
         ts_code  varchar(32) ,
         trade_date varchar(32),
         turnover_rate float,  
         volume_ratio float,
         pe float ,
         pb  float )   """
cursor.execute(sql)
db.close() # 关闭数据库连接
print('successful')

4、数据准备，需将dataframe数据转换为dict，参考了一篇不错的博文，链接放后面

####取原始数据，数据缺失用0补充，并将dataframe转换为dict字典类型，待后续使用，通过dict[s]取键为s对应的值
b=pro.query('daily_basic', ts_code='600230.SH', trade_date='20210519',fields='ts_code,trade_date,turnover_rate,volume_ratio,pe,pb').fillna(0)
type(b)
c=b.to_dict('index')
i=len(result.keys())
c=b.to_dict('index')[0