python量化投资（三）——从tushare读取股票信息（stock_list ）数据

最新推荐文章于 2024-09-13 14:35:02 发布

Joshua.Lee

最新推荐文章于 2024-09-13 14:35:02 发布

阅读量1.7k

点赞数 1

分类专栏：量化投资 python mysql 文章标签：数据库 mysql python 大数据数据挖掘

本文链接：https://blog.csdn.net/qq_21594155/article/details/110428305

版权

python 同时被 3 个专栏收录

10 篇文章 2 订阅

订阅专栏

量化投资

6 篇文章 2 订阅

订阅专栏

mysql

6 篇文章 0 订阅

订阅专栏

具体注册tushre，获取token不多在这里介绍了，官网对操作及接口描述很清晰
tushare官网地址tushare
从tushare返回的数据是pandas.DataFrame数据类型，为了能够将数据写入数据库，需要使用numpy将其转换成列表，逐行写入
使用numpy中的array方法将返回的数据表转换成多维数组，再使用tolist方法将多维数组转换成列表
代码如下：

import tushare as ts
import pymysql
import numpy as np

ts.set_token('你的token')

pro = ts.pro_api()

# 查询当前所有正常上市交易的股票列表
data = pro.stock_basic(exchange='', list_status='L',fields='ts_code,symbol,name,area,industry,fullname,enname,market,exchange,curr_type,list_status,list_date,delist_date,is_hs')
print(data)
dataset = np.array(data)
datalist = dataset.tolist()

# 打开数据库连接
db = pymysql.connect(host='localhost', port=3306, user='xxx', passwd='xxx', db='tushare', charset='utf8mb4')

# 使用cursor()方法获取操作游标
cursor = db.cursor()
sql = "delete from stock_list"
try:
    # 执行sql语句
    cursor.execute(sql)
    # 提交到数据库执行
    db.commit()
except Exception as e:
    print(e)
    # 如果发生错误则回滚
    db.rollback()
# 关闭游标
cursor.close()

cursor = db.cursor()
# SQL 插入语句
sql = "INSERT INTO stock_list (`ts_code`, `symbol`, `name`, `area`, `industry`, `fullname`, `enname`, `market`, `exchange`, `curr_type`, `list_status`, `list_date`, `delist_date`, `is_hs`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"

try:
    # 执行sql语句
    cursor.executemany(sql, datalist)
    # 提交到数据库执行
    db.commit()
except Exception as e:
    print(e)
    # 如果发生错误则回滚
    db.rollback()
# 关闭游标
cursor.close()
# 关闭数据库连接
db.close()