python利用tushare下载数据并计算当日收益率

最新推荐文章于 2023-03-30 18:10:28 发布

weixin_41755752

最新推荐文章于 2023-03-30 18:10:28 发布

阅读量2.2k

点赞数 2

分类专栏：量化投资文章标签： python 数据分析可视化

本文链接：https://blog.csdn.net/weixin_41755752/article/details/118912791

版权

本文介绍了如何使用Python结合tushare库下载股票数据，并详细阐述了计算股票当日收益率的程序，包括获取接口数据、计算收益率、可视化结果的函数，以及如何自定义建仓价格进行计算。通过实例展示了对推荐股票的收益率计算过程。

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

python利用tushare下载数据并计算当日收益率

计算股票收益率的程序主要有以下几部分构成：
1、获取股票接口数据函数：pro_daily_stock（）
2、计算收益率函数：cal_stock（），里面有两种计算式，
你可以根据自己字典写入建仓价格做差计算，也可以像我一样用收盘价-当天开盘价计算
3、可视化函数：画柱状图def draw_bar_stock（）
4、最后对象实例化，调用函数
里面有一个嵌套字典需要你手动填写的stock_sum={}
只填写股票代码和股票名称就够了，也可以填写你的买入价格。
以今天炒股机构给我推荐的10支股票为示例，
效果如下：
在这里插入图片描述

代码如下：

#! /usr/bin/env python
#-*- encoding: utf-8 -*-

"""
作者：廖超铭-量化交易员
仅用于教学目的，严禁转发和用于盈利目的，违者必究，若要引用，请注明出处
日期：2021-7-19
功能：股票收益可视化
"""
#导入模块
import matplotlib.pyplot as plt
import pandas as pd
import tushare as ts

#############参数
pd.set_option('display.expand_frame_repr',False)#False不允许换行
pd.set_option('display.max_rows', 10)#显示的最大行数
pd.set_option('display.max_columns', 6)#显示的最大列数
pd.set_option('precision', 1)#显示小数点后的位数
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号

##############函数部分
#一个获取股票价格的函数
def pro_daily_stock(code_val  ,start_val ,end_val):
    # 获取股票日线行情数据
    token = 'c3d13a70203e52cb7479f32953928b21680f94c255ff09aed868c4c4'
    pro = ts.pro_api(token)
    df_stock = pro.daily(ts_code=code_val,adj='qfq', start_date=start_val, end_date=end_val)#其中adj 是设置复权的接口，none：不复权  qfq：前复权  hfq：后复权
    df_stock.trade_date = pd.DatetimeIndex(df_stock.trade_date)
    df_stock.set_index("trade_date", drop=True, inplace=True)
    df_stock.sort_index(inplace=True)
    df_stock.index = df_stock.index.set_names('Date')

    recon_data = {
   'High': df_stock.high, 'Low': df_stock.low, 'Open': df_stock.open, 'Close': df_stock.close,\
                  'Volume': df_stock.vol}
    df_recon = pd.DataFrame(recon_data)

    return df_recon
# 定义一个函数用来计算收益率
def cal_stock(dict,starttime,endtime):
    #######获取字典数据部分
    name_stock = []
    code_stock =[]
    start_value =[]

    value = []
    for key in  dict:#最后应该组成一个表格，最后生成两个列表，index为股票名称，value为收益率
        n = 0<