import tushare as ts
import pandas as pd
import numpy as np
from scipy import stats
import tushare as ts
import matplotlib.pyplot as plt
%matplotlib inline #正常显示画图时出现的中文和负号
from pylab import mpl
mpl.rcParams['font.sans-serif']=['SimHei']
mpl.rcParams['axes.unicode_minus']=False
from datetime import datetime,timedelta
from pyecharts import *
token=''
ts.set_token(token)
pro=ts.pro_api()
获取数据
#获取数据并计算对数收益率def get_index_data(code):df=pd.DataFrame()df0=pro.index_daily(ts_code=code)df[code]=df0.closedf.index=pd.to_datetime(df0.trade_date)df=df.sort_index()df['logret']=np.log(df/df.shift(1))return df.iloc[1:,:]
#A股常用的八个指数
common_index={'上证综指': '000001.SH','深证成指': '399001.SZ','沪深300': '000300.SH','创业板指': '399006.SZ','上证50': '000016.SH','中证500': '000905.SH','中小板指': '399005.SZ','上证180': '000010.SH'}
#获取指数收盘价和日对数收益率
index_data={}
for name,code in common_index.items():index_data[name]=get_index_data(code)#对股价走势进行可视化
names =list(common_index.keys())
codes=list(common_index.values())plot_pos = [421,422,423,424,425,426,427,428] # 每个子图的位置
new_colors = ['#1f77b4','#ff7f0e', '#2ca02c', '#d62728', '#9467bd','#8c564b', '#e377c2', '#7f7f7f','#bcbd22','#17becf']fig = plt.figure(figsize=(16,18))
fig.suptitle('国内股指走势',fontsize=18)
for i in np.arange(len(plot_pos)): ax = fig.add_subplot(plot_pos[i]) y_data =index_data[names[i]][codes[i]] b = ax.plot(y_data,color=new_colors[i]) ax.set_title(names[i])
plt.show()
累计收益率
#股指自发行以来累积收益率情况
fig = plt.figure(figsize=(16,18))
fig.suptitle('国内股指累积收益率',fontsize=18)
for i in np.arange(len(plot_pos)): ax = fig.add_subplot(plot_pos[i]) y_data =index_data[names[i]]['logret'].cumsum() b = ax.plot(y_data,color=new_colors[i]) ax.set_title(names[i])
plt.show()
def cum_return(start_date,end_date='2018-12-24'):df=pd.DataFrame()for name,data in index_data.items():df[name]=data.loc[start_date:end_date]['logret'].cumsum()return df
区间收益率情况
cum_return('2013-01-01','2015-06-12').plot(figsize=(16,6))
cum_return('2011-01-01').plot(figsize=(16,6))
cum_return('2018-01-01').plot(figsize=(16,6))grid = Grid()
attr = list(common_index)
n=[1/28,1/27,1/13,1/8,1/14,1/13,1/12,1/22]
r=cum_return('1990-12-20').iloc[-1]
v1 = list((r.values*100).round(2))
v2 = ((np.power(1+r.values,n)-1)*100).round(2)bar = Bar(title="各指数收益率情况")
bar.add("累计收益率%", attr, v1,is_label_show=True)line = Line()
line.add("年化平均收益率%", attr, v2,is_label_show=True)
overlap = Overlap()
overlap.add(bar)
overlap.add(line, is_add_yaxis=True, yaxis_index=1)grid.add(overlap, grid_right="15%")
grid
指数定投评估
#指数定投函数
def index_strategy(code,start_date,end_date,first=1):##步骤一:获取数据##通过自定义函数get_index_data获取指数数据df=get_index_data(code)df=df.loc[start_date:end_date]#累计收益率df['累计收益率']=df.logret.cumsum()#假设定投无风险理财产品收益率为4%df['无风险收益率']=(4.0/100+1)**(1.0/250)-1df['无风险收益_净值']=(df['无风险收益率']+1).cumprod()##步骤二:设定定投日并计算份额和累计资金#每月第一个交易日或最后一个交易日定投#first==1表示每个月第一个交易日定投if first==1:by_month=df.resample('M',kind='period').first()else:by_month=df.resample('M',kind='period').last()#定投购买指数基金trade_log=pd.DataFrame(index=by_month.index)#假设每月投入3000元trade_log['基金净值']=by_month[code]/3000trade_log['money']=3000 trade_log['基金份额']=trade_log['money']/trade_log['基金净值']trade_log['总基金份额']=trade_log['基金份额'].cumsum()trade_log['累计定投资金']=trade_log['money'].cumsum()##步骤三:计算定投保本理财份额和资金trade_log['理财份额']=trade_log['money']/by_month['无风险收益_净值']trade_log['总理财份额']=trade_log['理财份额'].cumsum()temp=trade_log.resample('D').ffill()df=df.to_period('D')##步骤三:计算每个交易日的资产(等于每天的基金份额乘以单位基金净值)和累计收益率daily_data=pd.concat([df,temp[['总基金份额','总理财份额','累计定投资金']]],axis=1,join='inner')daily_data['指数定投资金']=daily_data[code]/3000*daily_data['总基金份额']daily_data['理财定投资金']=daily_data['无风险收益_净值']*daily_data['总理财份额']daily_data['指数定投收益率']=(daily_data['指数定投资金']-daily_data['累计定投资金'])/daily_data['累计定投资金']daily_data['理财定投收益率']=(daily_data['理财定投资金']-daily_data['累计定投资金'])/daily_data['累计定投资金']#返回后面分析需要的变量return daily_data[[code,'累计收益率','累计定投资金','指数定投资金','理财定投资金','指数定投收益率','理财定投收益率']]
方案1:
#将画图过程封装成函数
def plot_index_ret(code,name,start_date,end_date):df=index_strategy(code,start_date,end_date,first=1)df1=index_strategy(code,start_date,end_date,first=0)print(f'月初开始定投:\n{df.iloc[-1]}')print('\n')print(f'月末开始定投:{df1.iloc[-1][5]}')ax1=df[['累计收益率','指数定投收益率','理财定投收益率']].plot(figsize=(16,6))ax1.legend(loc=2)plt.title(name+'定投收益率情况\n '+start_date+':'+end_date,fontsize=15)ax2 = ax1.twinx()df[code].plot(figsize=(16,6),color='r',label=name)ax2.legend(loc=1)plt.show()def plot_index_fund(code,name,start_date,end_date):#每月定投3000元,期间累计资金情况df=index_strategy(code,start_date,end_date,first=1)ax1=df[['指数定投资金','理财定投资金']].plot(figsize=(16,6))ax2 = ax1.twinx()df[code].plot(figsize=(16,6),color='r',label=name)ax2.legend(loc=1)plt.title(f'累计定投资金情况\n {start_date}:{end_date}',fontsize=15)plt.show()
plot_index_ret('000001.SH','上证综指','2007-10-17','2015-6-12')attr = ["累计收益率 ","月初指数定投收益率","月底指数定投收益率","理财定投收益率"]
v1 = [-16.48, 101.03, 102.02, 16.28]
bar = Bar("上证综指定投收益率情况(%)","2007.10.17—2015.06.12'",title_text_size=15,title_pos='center')
bar.add("", attr, v1,is_label_show=True )
barplot_index_ret('000001.SH','上证综指','2007-10-17','2009-07-20')attr = ["累计收益率 ","月初指数定投收益率","月底指数定投收益率","理财定投收益率"]
v1 = [-62.31, 15.64, 17.96, 3.62]
bar = Bar("上证指数定投收益率情况(%)","2007.10-2009.9",title_text_size=15,title_pos='center')
bar.add("", attr, v1,is_label_show=True )
bar
方案2:
plot_index_ret('000001.SH','上证综指','2006-10-01','2018-12-24')attr = ["累计收益率 ","月初指数定投收益率","月底指数定投收益率","理财定投收益率"]
v1 = [36.6, -7.39, -7.46, 27.63]
bar = Bar("上证指数定投收益率情况(%)","2006.10-2018.12",title_text_size=15,title_pos='center')
bar.add("", attr, v1,is_label_show=True )
bar
方案3:
plot_index_ret('000001.SH','上证综指','2014-07-01','2015-06-12')plot_index_ret('399006.SZ','创业板指','2014-07-01','2015-06-12')#画柱状图函数
def plot_bar_ret(start_date,end_date):end_datev1=[]#定投累计收益率v2=[]for name,code in common_index.items():df=index_strategy(code,start_date,end_date,first=1)c=round((df['累计收益率'].iloc[-1])*100,2)a=round((df['指数定投收益率'].iloc[-1])*100,2)v1.append(c)v2.append(a)attr=list(common_index.keys())bar = Bar("各指数定投vs投入持有" ,f"{start_date}:{end_date}")bar.add("一次投累计收益率%", attr, v1, mark_point=["min", "max"])bar.add("定投累计收益率%", attr, v2, mark_point=["min", "max"])barstart_date='2014-07-01'
end_date='2015-06-12'
#一次性投累计收益率
v1=[]
#定投累计收益率
v2=[]
for name,code in common_index.items():df=index_strategy(code,start_date,end_date,first=1)c=round((df['累计收益率'].iloc[-1])*100,2)a=round((df['指数定投收益率'].iloc[-1])*100,2)v1.append(c)v2.append(a)
attr=list(common_index.keys())
bar = Bar("各指数定投累计收益率" ,"2014.07.01-2015.06.12")
bar.add("一次投入累计收益率%", attr, v1, mark_point=["min", "max"])
bar.add("定投累计收益率%", attr, v2, mark_point=["min", "max"])
bar
start_date='2011-08-01'
end_date='2016-12-24'
#一次性投累计收益率
v1=[]
#定投累计收益率
v2=[]
for name,code in common_index.items():df=index_strategy(code,start_date,end_date,first=1)c=round((df['累计收益率'].iloc[-1])*100,2)a=round((df['指数定投收益率'].iloc[-1])*100,2)v1.append(c)v2.append(a)
attr=list(common_index.keys())
bar = Bar("各指数定投累计收益率" ,"2011.08.01-2016.12.24")
bar.add("一次投入累计收益率%", attr, v1, mark_point=["min", "max"])
bar.add("定投累计收益率%", attr, v2, mark_point=["min", "max"])
bar
start_date='2011-08-01'
end_date='2018-12-24'
#一次性投累计收益率
v1=[]
#定投累计收益率
v2=[]
for name,code in common_index.items():df=index_strategy(code,start_date,end_date,first=1)c=round((df['累计收益率'].iloc[-1])*100,2)a=round((df['指数定投收益率'].iloc[-1])*100,2)v1.append(c)v2.append(a)
attr=list(common_index.keys())
bar = Bar("各指数定投累计收益率" ,"2011.08.01-2018.12.24")
bar.add("一次投入累计收益率%", attr, v1, mark_point=["min", "max"])
bar.add("定投累计收益率%", attr, v2, mark_point=["min", "max"])
bar
start_date='2015-06-12'
end_date='2018-12-24'
#一次性投累计收益率
v1=[]
#定投累计收益率
v2=[]
for name,code in common_index.items():df=index_strategy(code,start_date,end_date,first=1)c=round((df['累计收益率'].iloc[-1])*100,2)a=round((df['指数定投收益率'].iloc[-1])*100,2)v1.append(c)v2.append(a)
attr=list(common_index.keys())
bar = Bar("各指数定投累计收益率" ,"2015.06.12-2018.12.24")
bar.add("一次投入累计收益率%", attr, v1, mark_point=["min", "max"])
bar.add("定投累计收益率%", attr, v2, mark_point=["min", "max"])
bar
start_date='2018-01-01'
end_date='2018-12-24'
#一次性投累计收益率
v1=[]
#定投累计收益率
v2=[]
for name,code in common_index.items():df=index_strategy(code,start_date,end_date,first=1)c=round((df['累计收益率'].iloc[-1])*100,2)a=round((df['指数定投收益率'].iloc[-1])*100,2)v1.append(c)v2.append(a)
attr=list(common_index.keys())
bar = Bar("各指数定投收益率" ,"2018.01.01-2018.12.24")
bar.add("一次投入累计收益率%", attr, v1, mark_point=["min", "max"])
bar.add("定投累计收益率%", attr, v2, mark_point=["min", "max"])
bar