if name == ‘main’:
stocks = st.get_all_securities(‘stock’)
ywln_code = list(stocks[stocks.loc[:, ‘display_name’] == ‘亿纬锂能’].index)[0]
gzmt_code = list(stocks[stocks.loc[:, ‘display_name’] == ‘贵州茅台’].index)[0]
byd_code = list(stocks[stocks.loc[:, ‘display_name’] == ‘比亚迪’].index)[0]
ywln = week_period_strategy(ywln_code, ‘2019-1-1’, ‘2019-12-31’, ‘daily’)
gzmt = week_period_strategy(gzmt_code, ‘2019-1-1’, ‘2019-12-31’, ‘daily’)
byd = week_period_strategy(byd_code, ‘2019-1-1’, ‘2019-12-31’, ‘daily’)
mix = pd.DataFrame()
mix[‘ywln’] = ywln[‘cum_profit_pct’]
mix[‘gzmt’] = gzmt[‘cum_profit_pct’]
mix[‘byd’] = byd[‘cum_profit_pct’]
mix[[‘ywln’, ‘gzmt’, ‘byd’]].plot()
plt.show()