复制链接
克隆策略

    {"description":"实验创建于2017/8/26","graph":{"edges":[{"to_node_id":"-228:instruments","from_node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8:data"},{"to_node_id":"-234:instruments","from_node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8:data"},{"to_node_id":"-123:instruments","from_node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8:data"},{"to_node_id":"-550:input_1","from_node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8:data"},{"to_node_id":"-575:instruments","from_node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8:data"},{"to_node_id":"-228:features","from_node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-24:data"},{"to_node_id":"-235:features","from_node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-24:data"},{"to_node_id":"-235:input_data","from_node_id":"-228:data"},{"to_node_id":"-253:data1","from_node_id":"-234:data"},{"to_node_id":"-582:data1","from_node_id":"-253:data"},{"to_node_id":"-253:data2","from_node_id":"-235:data"},{"to_node_id":"-234:features","from_node_id":"-270:data"},{"to_node_id":"-123:options_data","from_node_id":"-185:data_1"},{"to_node_id":"-123:benchmark_ds","from_node_id":"-550:data_1"},{"to_node_id":"-575:features","from_node_id":"-570:data"},{"to_node_id":"-582:data2","from_node_id":"-575:data"},{"to_node_id":"-297:input_1","from_node_id":"-582:data"},{"to_node_id":"-185:input_1","from_node_id":"-297:data_1"},{"to_node_id":"-185:input_2","from_node_id":"-297:data_2"},{"to_node_id":"-185:input_3","from_node_id":"-297:data_3"}],"nodes":[{"node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8","module_id":"BigQuantSpace.instruments.instruments-v2","parameters":[{"name":"start_date","value":"2020-01-01","type":"Literal","bound_global_parameter":null},{"name":"end_date","value":"2022-12-16","type":"Literal","bound_global_parameter":null},{"name":"market","value":"CN_STOCK_A","type":"Literal","bound_global_parameter":null},{"name":"instrument_list","value":"","type":"Literal","bound_global_parameter":null},{"name":"max_count","value":"","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"rolling_conf","node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8"}],"output_ports":[{"name":"data","node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-8"}],"cacheable":true,"seq_num":1,"comment":"","comment_collapsed":true},{"node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-24","module_id":"BigQuantSpace.input_features.input_features-v1","parameters":[{"name":"features","value":"\nin_csi300_0\nin_csi500_0\nin_sse50_0\nindustry_sw_level1_0\nst_status_0\n\n# 选股条件\ncond1=(ta_kdj_j(high_0/adjust_factor_0, low_0/adjust_factor_0, close_0/adjust_factor_0, N=9, M1=3, M2=3)<-6)&\\\n(fs_net_profit_yoy_0>1.5)&\\\n(ta_macd(close_0/adjust_factor_0, fastperiod=6, slowperiod=38, signalperiod=6, derive='short'))\n\n# 排序选股\ncond2=1\n\n# 进场条件\ncond3=(ta_inverted_hammer(high_0/adjust_factor_0, low_0/adjust_factor_0, close_0/adjust_factor_0, open_0/adjust_factor_0))\n \n# 卖出条件\ncond4=1\n","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"features_ds","node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-24"}],"output_ports":[{"name":"data","node_id":"287d2cb0-f53c-4101-bdf8-104b137c8601-24"}],"cacheable":true,"seq_num":3,"comment":"","comment_collapsed":true},{"node_id":"-228","module_id":"BigQuantSpace.general_feature_extractor.general_feature_extractor-v7","parameters":[{"name":"start_date","value":"","type":"Literal","bound_global_parameter":null},{"name":"end_date","value":"","type":"Literal","bound_global_parameter":null},{"name":"before_start_days","value":"300","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"instruments","node_id":"-228"},{"name":"features","node_id":"-228"}],"output_ports":[{"name":"data","node_id":"-228"}],"cacheable":true,"seq_num":15,"comment":"","comment_collapsed":true},{"node_id":"-234","module_id":"BigQuantSpace.use_datasource.use_datasource-v1","parameters":[{"name":"datasource_id","value":"industry_CN_STOCK_A","type":"Literal","bound_global_parameter":null},{"name":"start_date","value":"","type":"Literal","bound_global_parameter":null},{"name":"end_date","value":"","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"instruments","node_id":"-234"},{"name":"features","node_id":"-234"}],"output_ports":[{"name":"data","node_id":"-234"}],"cacheable":true,"seq_num":5,"comment":"","comment_collapsed":true},{"node_id":"-253","module_id":"BigQuantSpace.join.join-v3","parameters":[{"name":"on","value":"date,instrument","type":"Literal","bound_global_parameter":null},{"name":"how","value":"inner","type":"Literal","bound_global_parameter":null},{"name":"sort","value":"False","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"data1","node_id":"-253"},{"name":"data2","node_id":"-253"}],"output_ports":[{"name":"data","node_id":"-253"}],"cacheable":true,"seq_num":7,"comment":"","comment_collapsed":true},{"node_id":"-235","module_id":"BigQuantSpace.derived_feature_extractor.derived_feature_extractor-v3","parameters":[{"name":"date_col","value":"date","type":"Literal","bound_global_parameter":null},{"name":"instrument_col","value":"instrument","type":"Literal","bound_global_parameter":null},{"name":"drop_na","value":"False","type":"Literal","bound_global_parameter":null},{"name":"remove_extra_columns","value":"False","type":"Literal","bound_global_parameter":null},{"name":"user_functions","value":"","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"input_data","node_id":"-235"},{"name":"features","node_id":"-235"}],"output_ports":[{"name":"data","node_id":"-235"}],"cacheable":true,"seq_num":16,"comment":"","comment_collapsed":true},{"node_id":"-270","module_id":"BigQuantSpace.input_features.input_features-v1","parameters":[{"name":"features","value":"concept\n","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"features_ds","node_id":"-270"}],"output_ports":[{"name":"data","node_id":"-270"}],"cacheable":true,"seq_num":10,"comment":"获取股票概念,并匹配选中的概念","comment_collapsed":false},{"node_id":"-123","module_id":"BigQuantSpace.trade.trade-v4","parameters":[{"name":"start_date","value":"","type":"Literal","bound_global_parameter":null},{"name":"end_date","value":"","type":"Literal","bound_global_parameter":null},{"name":"initialize","value":"\ndef prepare_index_data(context):\n \"\"\"准备指数数据\"\"\"\n if context.market_risk_conf != []:\n if len(context.market_risk_conf) == 1:\n index_code = context.market_risk_conf[0]['params']['index_code']\n start_date = '2015-01-01'\n end_date = context.end_date\n index_data = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code], start_date=start_date, end_date=end_date).set_index('date')\n \n if context.market_risk_conf[0]['method'] == 'market_ma_stoploss':\n ma_periods = int(context.market_risk_conf[0]['params']['ma_periods'])\n index_data['ma_%s'%ma_periods] = index_data['close'].rolling(ma_periods).mean()\n index_data['signal'] = np.where(index_data['close'] > index_data['ma_%s'%ma_periods], 'long', 'short')\n\n elif context.market_risk_conf[0]['method'] == 'market_fallrange_stoploss':\n days = context.market_risk_conf[0]['params']['days']\n fallrange = context.market_risk_conf[0]['params']['fallrange']\n index_data['signal'] = np.where(index_data['close']/index_data['close'].shift(days)-1 <= fallrange, 'long', 'short')\n context.index_signal_data = index_data \n \n if len(context.market_risk_conf) == 2:\n start_date = '2015-01-01'\n end_date = context.end_date \n if context.market_risk_conf[0]['method'] == 'market_ma_stoploss': \n index_code_1 = context.market_risk_conf[0]['params']['index_code']\n index_data_1 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_1], start_date=start_date, end_date=end_date).set_index('date')\n ma_periods = int(context.market_risk_conf[0]['params']['ma_periods'])\n \n index_code_2 = context.market_risk_conf[1]['params']['index_code']\n index_data_2 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_2], start_date=start_date, end_date=end_date).set_index('date')\n days = context.market_risk_conf[1]['params']['days']\n fallrange = context.market_risk_conf[1]['params']['fallrange']\n else:\n index_code_1 = context.market_risk_conf[1]['params']['index_code']\n index_data_1 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_1], start_date=start_date, end_date=end_date).set_index('date')\n ma_periods = int(context.market_risk_conf[1]['params']['ma_periods'])\n \n index_code_2 = context.market_risk_conf[0]['params']['index_code']\n index_data_2 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_2], start_date=start_date, end_date=end_date).set_index('date')\n days = context.market_risk_conf[0]['params']['days']\n fallrange = context.market_risk_conf[0]['params']['fallrange'] \n \n index_data_1['ma_%s'%ma_periods] = index_data_1['close'].rolling(ma_periods).mean()\n index_data_1['signal_1'] = np.where(index_data_1['close'] > index_data_1['ma_%s'%ma_periods], 1, 0)\n signal_1 = index_data_1[['signal_1']].reset_index() \n index_data_2['signal_2'] = np.where(index_data_2['close']/index_data_2['close'].shift(days)-1 <= fallrange, 1, 0)\n signal_2 = index_data_2[['signal_2']].reset_index()\n signal = pd.merge(signal_1,signal_2).set_index('date')\n signal['signal_sum'] = signal['signal_1'] + signal['signal_2']\n signal['signal'] = np.where(signal['signal_sum']>0,'long','short') \n context.index_signal_data = signal\n else:\n context.index_signal_data = None \n\ndef bigquant_run(context):\n context.set_commission(PerOrder(buy_cost=0.0035, sell_cost=0.0045, min_cost=5))\n context.selected_stock = []\n context.trade_mode = '择时'\n\n if context.trade_mode == '轮动':\n context.buy_frequency = 1\n context.sell_frequency = 1\n context.rebalance_periods = 1 # 调仓周期\n context.max_stock_count = 5 # 最大持仓股票数量\n context.order_weight_method = 'equal_weight' # 买入方式\n context.is_sell_willbuy_stock = False # 卖出欲买进股票 \n else:\n # 买入条件参数\n context.stock_select_frequency = 1 # 选股频率\n context.order_weight_method = 'equal_weight' # 买入方式\n context.buy_frequency = 2 # 买入频率\n context.can_duplication_buy = False # 是否可重复买入\n context.max_stock_count = 8 # 最大持仓股票数量\n context.max_stock_weight = 0.1 # 个股最大持仓比重\n\n # 卖出条件参数\n context.sell_frequency = 10 # 卖出频率\n context.is_sell_willbuy_stock = False # 卖出欲买进股票 \n\n # 风控参数 \n context.stock_risk_conf = [{'method':'stock_percent_stopwin', 'params':{'percent': 0.08}}, {'method':'stock_percent_stoploss', 'params':{'percent': 0.04}}] # 支持多选 无:[]\n context.strategy_risk_conf = [] # 支持多选 无:[]\n context.market_risk_conf = [] # 支持多选, 无: []\n \n prepare_index_data(context)\n slippage_type = 'price'\n from zipline.finance.slippage import SlippageModel\n class FixedPriceSlippage(SlippageModel):\n # 指定初始化函数\n def __init__(self, spreads, price_field_buy, price_field_sell):\n # 存储spread的字典,用股票代码作为key\n self.spreads = spreads\n self._price_field_buy = price_field_buy\n self._price_field_sell = price_field_sell\n def process_order(self, data, order, bar_volume=0, trigger_check_price=0):\n if order.limit is None:\n price_field = self._price_field_buy if order.amount > 0 else self._price_field_sell\n price_base = data.current(order.asset, price_field)\n if slippage_type == 'price':\n price = price_base + (self.spreads / 2) if order.amount > 0 else price_base - (self.spreads / 2)\n else:\n price = price_base * (1.0 + self.spreads / 2) if order.amount > 0 else price_base * (1.0 - self.spreads / 2)\n else:\n price = order.limit\n # 返回希望成交的价格和数量\n return (price, order.amount)\n # 设置price_field\n fix_slippage = FixedPriceSlippage(price_field_buy='open', price_field_sell='open', spreads=0.03)\n context.set_slippage(us_equities=fix_slippage)","type":"Literal","bound_global_parameter":null},{"name":"handle_data","value":"#--------------------------------------------------------------------\n# 卖出条件\n#-------------------------------------------------------------------- \ndef sell_action(context, data):\n date = data.current_dt.strftime('%Y-%m-%d')\n hit_stop_stock = context.stock_hit_stop \n \n try:\n today_enter_stock = context.enter_daily_df.loc[date] \n except KeyError as e:\n today_enter_stock = []\n try:\n today_exit_stock = context.exit_daily_df.loc[date] \n except KeyError as e:\n today_exit_stock = []\n \n target_stock_to_buy = [i for i in context.selected_stock if i in today_enter_stock ] \n stock_hold_now = [equity.symbol for equity in context.portfolio.positions] # 当前持仓股票\n \n if context.trading_day_index % context.sell_frequency == 0:\n stock_to_sell = [i for i in stock_hold_now if i in today_exit_stock] # 要卖出的股票\n stock_buy_and_sell = [i for i in stock_to_sell if i in target_stock_to_buy]\n if context.is_sell_willbuy_stock == False: # 要买入的股票不卖出,但该票也不再买入\n stock_to_sell.extend(hit_stop_stock) # 将触发个股风控的股票融入到卖出票池\n stock_to_sell = [i for i in stock_to_sell if i not in stock_buy_and_sell] # 进行更新而已\n elif context.is_sell_willbuy_stock == True: # 要买入的股票依然要卖出,该票不再买入\n stock_to_sell.extend(hit_stop_stock)\n \n # 买入时需要过滤的股票\n context.cannot_buy_stock = stock_buy_and_sell\n \n for stock in stock_to_sell:\n if data.can_trade(context.symbol(stock)):\n context.order_target_percent(context.symbol(stock), 0)\n del context.portfolio.positions[context.symbol(stock)]\n\n\n#--------------------------------------------------------------------\n# 买入条件\n#-------------------------------------------------------------------- \ndef buy_action(context, data):\n date = data.current_dt.strftime('%Y-%m-%d')\n \n try:\n today_enter_stock = context.enter_daily_df.loc[date] \n except KeyError as e:\n today_enter_stock = []\n try:\n today_exit_stock = context.exit_daily_df.loc[date] \n except KeyError as e:\n today_exit_stock = []\n \n target_stock_to_buy = [i for i in context.selected_stock if i in today_enter_stock] \n target_stock_to_buy = [s for s in target_stock_to_buy if s not in context.cannot_buy_stock] # 进行更新,不能买入的股票要过滤\n \n stock_hold_now = [equity.symbol for equity in context.portfolio.positions] # 当前持仓股票\n \n # 确定股票权重\n if context.order_weight_method == 'equal_weight':\n equal_weight = 1 / context.max_stock_count\n \n portfolio_value = context.portfolio.portfolio_value\n position_current_value = {pos.sid: pos.amount* pos.last_sale_price for i,pos in context.portfolio.positions.items()}\n \n # 买入\n if context.trading_day_index % context.buy_frequency == 0:\n if len(stock_hold_now) >= context.max_stock_count:\n return \n \n today_buy_count = 0\n if context.trade_mode == '轮动':\n for s in target_stock_to_buy:\n if today_buy_count + len(stock_hold_now) >= context.max_stock_count: # 超出最大持仓数量\n break\n if data.can_trade(context.symbol(s)):\n order_target_percent(context.symbol(s), equal_weight)\n today_buy_count += 1\n else:\n if context.can_duplication_buy == True: # 可以重复买入,多一份买入\n for s in target_stock_to_buy:\n if today_buy_count + len(stock_hold_now) >= context.max_stock_count: # 超出最大持仓数量\n break\n \n if data.can_trade(context.symbol(s)):\n if context.symbol(s) in position_current_weight:\n curr_value = position_current_value.get(context.symbol(s)) \n order_value(context.symbol(s), min(context.max_stock_weight * portfolio_value - curr_value, equal_weight*portfolio_value))\n else:\n order_value(context.symbol(s), equal_weight*portfolio_value)\n today_buy_count += 1\n\n elif context.can_duplication_buy == False: # 不可以重复买入,不买\n for s in target_stock_to_buy:\n if today_buy_count + len(stock_hold_now) >= context.max_stock_count: # 超出最大持仓数量\n break\n if s in stock_hold_now:\n continue\n else:\n if data.can_trade(context.symbol(s)):\n order_target_percent(context.symbol(s), equal_weight)\n today_buy_count += 1\n\n \n#--------------------------------------------------------------------\n# 风控体系\n#-------------------------------------------------------------------- \ndef market_risk_manage(context, data):\n \"\"\"大盘风控\"\"\"\n date = data.current_dt.strftime('%Y-%m-%d')\n if type(context.index_signal_data) == pd.DataFrame:\n current_signal = context.index_signal_data.loc[date]['signal']\n if current_signal == 'short': \n stock_hold_now = [equity.symbol for equity in context.portfolio.positions] \n # 平掉所有股票\n for stock in stock_hold_now:\n if data.can_trade(context.symbol(stock)):\n context.order_target_percent(context.symbol(stock), 0) \n print('大盘出现止损信号, 平掉全部仓位,并关闭交易!')\n context.market_risk_signal = 'short'\n else:\n context.market_risk_signal = 'long'\n\n \n \ndef strategy_risk_manage(context, data):\n \"\"\"策略风控\"\"\"\n if context.strategy_risk_conf == []: # 没有设置策略风控\n context.strategy_risk_signal = 'long'\n \n else:\n for rm in context.strategy_risk_conf:\n if rm['method'] == 'strategy_percent_stopwin':\n pct = rm['params']['percent']\n portfolio_value = context.portfolio.portfolio_value \n if portfolio_value / context.capital_base - 1 > pct: \n stock_hold_now = [equity.symbol for equity in context.portfolio.positions] \n # 平掉所有股票\n for stock in stock_hold_now:\n if data.can_trade(context.symbol(stock)):\n context.order_target_percent(context.symbol(stock), 0) \n print('策略出现止盈信号, 平掉全部仓位,并关闭交易!')\n context.strategy_risk_signal = 'short' \n \n \n if rm['method'] == 'strategy_percent_stoploss':\n pct = rm['params']['percent']\n portfolio_value = context.portfolio.portfolio_value \n if portfolio_value / context.capital_base -1 < pct:\n stock_hold_now = [equity.symbol for equity in context.portfolio.positions] \n # 平掉所有股票\n for stock in stock_hold_now:\n if data.can_trade(context.symbol(stock)):\n context.order_target_percent(context.symbol(stock), 0) \n print('策略出现止损信号, 平掉全部仓位,并关闭交易!')\n context.strategy_risk_signal = 'short'\n\n \ndef stock_risk_manage(context, data):\n \"\"\"个股风控\"\"\"\n position_current_pnl = {pos.sid: (pos.last_sale_price-pos.cost_basis)/pos.cost_basis for i,pos in context.portfolio.positions.items()}\n \n for rm in context.stock_risk_conf:\n params_pct = rm['params']['percent']\n if rm['method'] == 'stock_percent_stopwin':\n for sid,pnl_pct in position_current_pnl.items(): \n if pnl_pct > params_pct:\n context.stock_hit_stop.append(sid.symbol)\n \n if rm['method'] == 'stock_percent_stoploss':\n for sid,pnl_pct in position_current_pnl.items():\n if pnl_pct < params_pct:\n context.stock_hit_stop.append(sid.symbol)\n\n\n\n\n# 回测引擎:每日数据处理函数,每天执行一次\ndef bigquant_run(context, data):\n \"\"\"每日运行策略逻辑\"\"\"\n market_risk_manage(context, data)\n strategy_risk_manage(context, data)\n \n if context.market_risk_signal == 'short': return\n if context.strategy_risk_signal == 'short': return\n\n stock_risk_manage(context, data)\n \n sell_action(context, data)\n buy_action(context, data)\n","type":"Literal","bound_global_parameter":null},{"name":"prepare","value":"# 回测引擎:准备数据,只执行一次\ndef bigquant_run(context):\n \n load_data = context.options['data'].read_pickle()\n context.signal_daily_stock = load_data['df1'].groupby('date').apply(lambda x:list(x.instrument))\n context.enter_daily_df = load_data['df2'].groupby('date').apply(lambda x:list(x.instrument))\n context.exit_daily_df = load_data['df3'].groupby('date').apply(lambda x:list(x.instrument))\n","type":"Literal","bound_global_parameter":null},{"name":"before_trading_start","value":"# 回测引擎:每个单位时间开始前调用一次,即每日开盘前调用一次。\ndef bigquant_run(context, data):\n \n \"\"\"每日盘前更新股票池\"\"\"\n frequency = context.rebalance_periods if context.trade_mode == '轮动' else context.stock_select_frequency\n if context.trading_day_index % frequency == 0:\n date = data.current_dt.strftime('%Y-%m-%d')\n try:\n context.selected_stock = context.signal_daily_stock[date] \n except KeyError as e:\n context.selected_stock = []\n \n \"\"\"初始化风控参数\"\"\"\n context.strategy_risk_signal = 'long'\n context.market_risk_signal = 'long' \n context.stock_hit_stop = []\n context.cannot_buy_stock = []\n","type":"Literal","bound_global_parameter":null},{"name":"volume_limit","value":0.025,"type":"Literal","bound_global_parameter":null},{"name":"order_price_field_buy","value":"open","type":"Literal","bound_global_parameter":null},{"name":"order_price_field_sell","value":"open","type":"Literal","bound_global_parameter":null},{"name":"capital_base","value":"30000","type":"Literal","bound_global_parameter":null},{"name":"auto_cancel_non_tradable_orders","value":"True","type":"Literal","bound_global_parameter":null},{"name":"data_frequency","value":"daily","type":"Literal","bound_global_parameter":null},{"name":"price_type","value":"真实价格","type":"Literal","bound_global_parameter":null},{"name":"product_type","value":"股票","type":"Literal","bound_global_parameter":null},{"name":"plot_charts","value":"True","type":"Literal","bound_global_parameter":null},{"name":"backtest_only","value":"False","type":"Literal","bound_global_parameter":null},{"name":"benchmark","value":"000300.SHA","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"instruments","node_id":"-123"},{"name":"options_data","node_id":"-123"},{"name":"history_ds","node_id":"-123"},{"name":"benchmark_ds","node_id":"-123"},{"name":"trading_calendar","node_id":"-123"}],"output_ports":[{"name":"raw_perf","node_id":"-123"}],"cacheable":false,"seq_num":4,"comment":"","comment_collapsed":true},{"node_id":"-185","module_id":"BigQuantSpace.cached.cached-v3","parameters":[{"name":"run","value":"# Python 代码入口函数,input_1/2/3 对应三个输入端,data_1/2/3 对应三个输出端\ndef bigquant_run(input_1, input_2, input_3):\n df1 = input_1.read_df()\n df2 = input_2.read_df()\n df3 = input_3.read_df()\n\n if len(df1.index.names) == 2:\n df1.index.names = [None, None]\n else:\n df1.index.names = [None]\n \n df = {'df1':df1,'df2':df2,'df3':df3}\n ds = DataSource.write_pickle(df)\n return Outputs(data_1=ds)\n","type":"Literal","bound_global_parameter":null},{"name":"post_run","value":"# 后处理函数,可选。输入是主函数的输出,可以在这里对数据做处理,或者返回更友好的outputs数据格式。此函数输出不会被缓存。\ndef bigquant_run(outputs):\n return outputs\n","type":"Literal","bound_global_parameter":null},{"name":"input_ports","value":"","type":"Literal","bound_global_parameter":null},{"name":"params","value":"{}","type":"Literal","bound_global_parameter":null},{"name":"output_ports","value":"","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"input_1","node_id":"-185"},{"name":"input_2","node_id":"-185"},{"name":"input_3","node_id":"-185"}],"output_ports":[{"name":"data_1","node_id":"-185"},{"name":"data_2","node_id":"-185"},{"name":"data_3","node_id":"-185"}],"cacheable":true,"seq_num":17,"comment":"","comment_collapsed":true},{"node_id":"-550","module_id":"BigQuantSpace.cached.cached-v3","parameters":[{"name":"run","value":"# Python 代码入口函数,input_1/2/3 对应三个输入端,data_1/2/3 对应三个输出端\ndef bigquant_run(input_1, input_index):\n # 示例代码如下。在这里编写您的代码\n start_date=input_1.read_pickle()['start_date']\n end_date=input_1.read_pickle()['end_date']\n df = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[input_index],start_date=start_date,end_date=end_date,fields=['close'])\n data_1 = DataSource.write_df(df)\n return Outputs(data_1=data_1, data_2=None, data_3=None)\n","type":"Literal","bound_global_parameter":null},{"name":"post_run","value":"# 后处理函数,可选。输入是主函数的输出,可以在这里对数据做处理,或者返回更友好的outputs数据格式。此函数输出不会被缓存。\ndef bigquant_run(outputs):\n return outputs\n","type":"Literal","bound_global_parameter":null},{"name":"input_ports","value":"input_1","type":"Literal","bound_global_parameter":null},{"name":"params","value":"{'input_index':'000300.HIX'}","type":"Literal","bound_global_parameter":null},{"name":"output_ports","value":"data_1","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"input_1","node_id":"-550"},{"name":"input_2","node_id":"-550"},{"name":"input_3","node_id":"-550"}],"output_ports":[{"name":"data_1","node_id":"-550"},{"name":"data_2","node_id":"-550"},{"name":"data_3","node_id":"-550"}],"cacheable":true,"seq_num":8,"comment":"","comment_collapsed":true},{"node_id":"-570","module_id":"BigQuantSpace.input_features.input_features-v1","parameters":[{"name":"features","value":"suspended","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"features_ds","node_id":"-570"}],"output_ports":[{"name":"data","node_id":"-570"}],"cacheable":true,"seq_num":6,"comment":"获取股票停牌数据","comment_collapsed":false},{"node_id":"-575","module_id":"BigQuantSpace.use_datasource.use_datasource-v1","parameters":[{"name":"datasource_id","value":"stock_status_CN_STOCK_A","type":"Literal","bound_global_parameter":null},{"name":"start_date","value":"","type":"Literal","bound_global_parameter":null},{"name":"end_date","value":"","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"instruments","node_id":"-575"},{"name":"features","node_id":"-575"}],"output_ports":[{"name":"data","node_id":"-575"}],"cacheable":true,"seq_num":19,"comment":"","comment_collapsed":true},{"node_id":"-582","module_id":"BigQuantSpace.join.join-v3","parameters":[{"name":"on","value":"date,instrument","type":"Literal","bound_global_parameter":null},{"name":"how","value":"inner","type":"Literal","bound_global_parameter":null},{"name":"sort","value":"False","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"data1","node_id":"-582"},{"name":"data2","node_id":"-582"}],"output_ports":[{"name":"data","node_id":"-582"}],"cacheable":true,"seq_num":20,"comment":"","comment_collapsed":true},{"node_id":"-297","module_id":"BigQuantSpace.cached.cached-v3","parameters":[{"name":"run","value":"# Python 代码入口函数,input_1/2/3 对应三个输入端,data_1/2/3 对应三个输出端\ndef bigquant_run(input_1, input_2, input_3):\n # 示例代码如下。在这里编写您的代码\n df = input_1.read_df()\n # 缺失值处理\n # if len(df)!=0:\n # df.dropna(inplace=True)\n \n # 选股条件\n if len(df)!=0:\n df_filter1 = df[df['cond1']>0]\n else:\n df_filter1 = df\n \n # 指标排序\n if len(df_filter1)!=0:\n df_filter2 = df_filter1.groupby('date').apply(lambda x:x.sort_values(by=['cond2'],ascending=True))\n else:\n df_filter2 = df_filter1\n \n #输出条件过滤股票池\n data_1 = DataSource.write_df(df_filter2)\n\n \n # 进场条件\n if len(df)!=0:\n df_buy = df[df['cond3']>0]\n else:\n df_buy = df\n # 输出满足进场条件的股票池\n data_2 = DataSource.write_df(df_buy)\n\n \n # 出场条件\n if len(df)!=0:\n df_sell = df[df['cond4']>0]\n else:\n df_sell = df\n # 输出满足出场条件的股票池\n data_3 = DataSource.write_df(df_sell) \n \n return Outputs(data_1=data_1, data_2=data_2, data_3=data_3)\n","type":"Literal","bound_global_parameter":null},{"name":"post_run","value":"# 后处理函数,可选。输入是主函数的输出,可以在这里对数据做处理,或者返回更友好的outputs数据格式。此函数输出不会被缓存。\ndef bigquant_run(outputs):\n return outputs\n","type":"Literal","bound_global_parameter":null},{"name":"input_ports","value":"","type":"Literal","bound_global_parameter":null},{"name":"params","value":"{}","type":"Literal","bound_global_parameter":null},{"name":"output_ports","value":"","type":"Literal","bound_global_parameter":null}],"input_ports":[{"name":"input_1","node_id":"-297"},{"name":"input_2","node_id":"-297"},{"name":"input_3","node_id":"-297"}],"output_ports":[{"name":"data_1","node_id":"-297"},{"name":"data_2","node_id":"-297"},{"name":"data_3","node_id":"-297"}],"cacheable":true,"seq_num":22,"comment":"","comment_collapsed":true}],"node_layout":"<node_postions><node_position Node='287d2cb0-f53c-4101-bdf8-104b137c8601-8' Position='473,-148,200,200'/><node_position Node='287d2cb0-f53c-4101-bdf8-104b137c8601-24' Position='697,-231,200,200'/><node_position Node='-228' Position='479,-22,200,200'/><node_position Node='-234' Position='145,-30,200,200'/><node_position Node='-253' Position='341,161,200,200'/><node_position Node='-235' Position='483,70,200,200'/><node_position Node='-270' Position='124,-149,200,200'/><node_position Node='-123' Position='563,687,200,200'/><node_position Node='-185' Position='346.31634521484375,521.1581420898438,200,200'/><node_position Node='-550' Position='864.9473876953125,403.31634521484375,200,200'/><node_position Node='-570' Position='960,-137,200,200'/><node_position Node='-575' Position='840,-16,200,200'/><node_position Node='-582' Position='374.2640380859375,291.5787048339844,200,200'/><node_position Node='-297' Position='353.47418212890625,420.73699951171875,200,200'/></node_postions>"},"nodes_readonly":false,"studio_version":"v2"}
    In [1]:
    # 本代码由可视化策略环境自动生成 2022年12月28日 08:28
    # 本代码单元只能在可视化模式下编辑。您也可以拷贝代码,粘贴到新建的代码单元或者策略,然后修改。
    
    
    # Python 代码入口函数,input_1/2/3 对应三个输入端,data_1/2/3 对应三个输出端
    def m8_run_bigquant_run(input_1, input_index):
        # 示例代码如下。在这里编写您的代码
        start_date=input_1.read_pickle()['start_date']
        end_date=input_1.read_pickle()['end_date']
        df = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[input_index],start_date=start_date,end_date=end_date,fields=['close'])
        data_1 = DataSource.write_df(df)
        return Outputs(data_1=data_1, data_2=None, data_3=None)
    
    # 后处理函数,可选。输入是主函数的输出,可以在这里对数据做处理,或者返回更友好的outputs数据格式。此函数输出不会被缓存。
    def m8_post_run_bigquant_run(outputs):
        return outputs
    
    # Python 代码入口函数,input_1/2/3 对应三个输入端,data_1/2/3 对应三个输出端
    def m22_run_bigquant_run(input_1, input_2, input_3):
        # 示例代码如下。在这里编写您的代码
        df = input_1.read_df()
        # 缺失值处理
        # if len(df)!=0:
        #     df.dropna(inplace=True)
            
        # 选股条件
        if len(df)!=0:
            df_filter1 = df[df['cond1']>0]
        else:
            df_filter1 = df
        
        # 指标排序
        if len(df_filter1)!=0:
            df_filter2 = df_filter1.groupby('date').apply(lambda x:x.sort_values(by=['cond2'],ascending=True))
        else:
            df_filter2 = df_filter1
        
        #输出条件过滤股票池
        data_1 = DataSource.write_df(df_filter2)
    
        
        # 进场条件
        if len(df)!=0:
            df_buy = df[df['cond3']>0]
        else:
            df_buy = df
        # 输出满足进场条件的股票池
        data_2 = DataSource.write_df(df_buy)
    
        
        # 出场条件
        if len(df)!=0:
            df_sell = df[df['cond4']>0]
        else:
            df_sell = df
        # 输出满足出场条件的股票池
        data_3 = DataSource.write_df(df_sell)    
        
        return Outputs(data_1=data_1, data_2=data_2, data_3=data_3)
    
    # 后处理函数,可选。输入是主函数的输出,可以在这里对数据做处理,或者返回更友好的outputs数据格式。此函数输出不会被缓存。
    def m22_post_run_bigquant_run(outputs):
        return outputs
    
    # Python 代码入口函数,input_1/2/3 对应三个输入端,data_1/2/3 对应三个输出端
    def m17_run_bigquant_run(input_1, input_2, input_3):
        df1 = input_1.read_df()
        df2 = input_2.read_df()
        df3 = input_3.read_df()
    
        if len(df1.index.names) == 2:
            df1.index.names = [None, None]
        else:
            df1.index.names = [None]
        
        df = {'df1':df1,'df2':df2,'df3':df3}
        ds = DataSource.write_pickle(df)
        return Outputs(data_1=ds)
    
    # 后处理函数,可选。输入是主函数的输出,可以在这里对数据做处理,或者返回更友好的outputs数据格式。此函数输出不会被缓存。
    def m17_post_run_bigquant_run(outputs):
        return outputs
    
    
    def prepare_index_data(context):
        """准备指数数据"""
        if context.market_risk_conf != []:
            if len(context.market_risk_conf) == 1:
                index_code =  context.market_risk_conf[0]['params']['index_code']
                start_date = '2015-01-01'
                end_date = context.end_date
                index_data = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code], start_date=start_date, end_date=end_date).set_index('date')
                
                if  context.market_risk_conf[0]['method'] == 'market_ma_stoploss':
                    ma_periods = int(context.market_risk_conf[0]['params']['ma_periods'])
                    index_data['ma_%s'%ma_periods] = index_data['close'].rolling(ma_periods).mean()
                    index_data['signal'] = np.where(index_data['close'] > index_data['ma_%s'%ma_periods], 'long', 'short')
    
                elif context.market_risk_conf[0]['method'] == 'market_fallrange_stoploss':
                    days = context.market_risk_conf[0]['params']['days']
                    fallrange = context.market_risk_conf[0]['params']['fallrange']
                    index_data['signal'] = np.where(index_data['close']/index_data['close'].shift(days)-1 <= fallrange, 'long', 'short')
                context.index_signal_data = index_data 
                
            if  len(context.market_risk_conf) == 2:
                start_date = '2015-01-01'
                end_date = context.end_date 
                if context.market_risk_conf[0]['method'] == 'market_ma_stoploss':              
                    index_code_1 = context.market_risk_conf[0]['params']['index_code']
                    index_data_1 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_1], start_date=start_date, end_date=end_date).set_index('date')
                    ma_periods = int(context.market_risk_conf[0]['params']['ma_periods'])
                    
                    index_code_2 =  context.market_risk_conf[1]['params']['index_code']
                    index_data_2 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_2], start_date=start_date, end_date=end_date).set_index('date')
                    days = context.market_risk_conf[1]['params']['days']
                    fallrange = context.market_risk_conf[1]['params']['fallrange']
                else:
                    index_code_1 = context.market_risk_conf[1]['params']['index_code']
                    index_data_1 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_1], start_date=start_date, end_date=end_date).set_index('date')
                    ma_periods = int(context.market_risk_conf[1]['params']['ma_periods'])
                    
                    index_code_2 =  context.market_risk_conf[0]['params']['index_code']
                    index_data_2 = DataSource('bar1d_index_CN_STOCK_A').read(instruments=[index_code_2], start_date=start_date, end_date=end_date).set_index('date')
                    days = context.market_risk_conf[0]['params']['days']
                    fallrange = context.market_risk_conf[0]['params']['fallrange'] 
                    
                index_data_1['ma_%s'%ma_periods] = index_data_1['close'].rolling(ma_periods).mean()
                index_data_1['signal_1'] = np.where(index_data_1['close'] > index_data_1['ma_%s'%ma_periods], 1, 0)
                signal_1 = index_data_1[['signal_1']].reset_index()                
                index_data_2['signal_2'] = np.where(index_data_2['close']/index_data_2['close'].shift(days)-1 <= fallrange, 1, 0)
                signal_2 = index_data_2[['signal_2']].reset_index()
                signal = pd.merge(signal_1,signal_2).set_index('date')
                signal['signal_sum'] = signal['signal_1'] + signal['signal_2']
                signal['signal'] = np.where(signal['signal_sum']>0,'long','short')     
                context.index_signal_data = signal
        else:
            context.index_signal_data = None  
    
    def m4_initialize_bigquant_run(context):
        context.set_commission(PerOrder(buy_cost=0.0035, sell_cost=0.0045, min_cost=5))
        context.selected_stock = []
        context.trade_mode = '择时'
    
        if context.trade_mode == '轮动':
            context.buy_frequency = 1
            context.sell_frequency = 1
            context.rebalance_periods = 1 # 调仓周期
            context.max_stock_count = 5 # 最大持仓股票数量
            context.order_weight_method = 'equal_weight' # 买入方式
            context.is_sell_willbuy_stock = False  # 卖出欲买进股票 
        else:
            # 买入条件参数
            context.stock_select_frequency = 1 # 选股频率
            context.order_weight_method = 'equal_weight' # 买入方式
            context.buy_frequency = 2 # 买入频率
            context.can_duplication_buy = False  # 是否可重复买入
            context.max_stock_count = 8 # 最大持仓股票数量
            context.max_stock_weight = 0.1 # 个股最大持仓比重
    
            # 卖出条件参数
            context.sell_frequency = 10 # 卖出频率
            context.is_sell_willbuy_stock = False  # 卖出欲买进股票 
    
        # 风控参数 
        context.stock_risk_conf = [{'method':'stock_percent_stopwin', 'params':{'percent': 0.08}}, {'method':'stock_percent_stoploss', 'params':{'percent': 0.04}}] # 支持多选  无:[]
        context.strategy_risk_conf = []  # 支持多选 无:[]
        context.market_risk_conf = [] # 支持多选, 无: []
        
        prepare_index_data(context)
        slippage_type = 'price'
        from zipline.finance.slippage import SlippageModel
        class FixedPriceSlippage(SlippageModel):
            # 指定初始化函数
            def __init__(self, spreads, price_field_buy, price_field_sell):
                # 存储spread的字典,用股票代码作为key
                self.spreads = spreads
                self._price_field_buy = price_field_buy
                self._price_field_sell = price_field_sell
            def process_order(self, data, order, bar_volume=0, trigger_check_price=0):
                if order.limit is None:
                    price_field = self._price_field_buy if order.amount > 0 else self._price_field_sell
                    price_base = data.current(order.asset, price_field)
                    if slippage_type == 'price':
                        price = price_base + (self.spreads / 2) if order.amount > 0 else price_base - (self.spreads / 2)
                    else:
                        price = price_base * (1.0 + self.spreads / 2) if order.amount > 0 else price_base * (1.0 - self.spreads / 2)
                else:
                    price = order.limit
                    # 返回希望成交的价格和数量
                return (price, order.amount)
        # 设置price_field
        fix_slippage = FixedPriceSlippage(price_field_buy='open', price_field_sell='open', spreads=0.03)
        context.set_slippage(us_equities=fix_slippage)
    #--------------------------------------------------------------------
    # 卖出条件
    #--------------------------------------------------------------------     
    def sell_action(context, data):
        date = data.current_dt.strftime('%Y-%m-%d')
        hit_stop_stock = context.stock_hit_stop 
        
        try:
            today_enter_stock = context.enter_daily_df.loc[date] 
        except KeyError as e:
            today_enter_stock = []
        try:
            today_exit_stock = context.exit_daily_df.loc[date] 
        except KeyError as e:
            today_exit_stock = []
            
        target_stock_to_buy = [i for i in  context.selected_stock  if i in today_enter_stock ]   
        stock_hold_now = [equity.symbol for equity in context.portfolio.positions] # 当前持仓股票
        
        if  context.trading_day_index % context.sell_frequency == 0:
            stock_to_sell = [i for i in stock_hold_now if i in today_exit_stock] # 要卖出的股票
            stock_buy_and_sell = [i for i in stock_to_sell if i in target_stock_to_buy]
            if context.is_sell_willbuy_stock == False: # 要买入的股票不卖出,但该票也不再买入
                stock_to_sell.extend(hit_stop_stock) # 将触发个股风控的股票融入到卖出票池
                stock_to_sell = [i for i in stock_to_sell if i not in stock_buy_and_sell] # 进行更新而已
            elif context.is_sell_willbuy_stock == True: # 要买入的股票依然要卖出,该票不再买入
                stock_to_sell.extend(hit_stop_stock)
            
            # 买入时需要过滤的股票
            context.cannot_buy_stock = stock_buy_and_sell
                
            for stock in stock_to_sell:
                if data.can_trade(context.symbol(stock)):
                    context.order_target_percent(context.symbol(stock), 0)
                    del context.portfolio.positions[context.symbol(stock)]
    
    
    #--------------------------------------------------------------------
    # 买入条件
    #--------------------------------------------------------------------     
    def buy_action(context, data):
        date = data.current_dt.strftime('%Y-%m-%d')
        
        try:
            today_enter_stock = context.enter_daily_df.loc[date] 
        except KeyError as e:
            today_enter_stock = []
        try:
            today_exit_stock = context.exit_daily_df.loc[date] 
        except KeyError as e:
            today_exit_stock = []
        
        target_stock_to_buy = [i for i in  context.selected_stock if i in today_enter_stock]  
        target_stock_to_buy = [s for s in target_stock_to_buy if s not in context.cannot_buy_stock] # 进行更新,不能买入的股票要过滤
        
        stock_hold_now = [equity.symbol for equity in context.portfolio.positions] # 当前持仓股票
        
        # 确定股票权重
        if context.order_weight_method == 'equal_weight':
            equal_weight =  1 / context.max_stock_count
            
        portfolio_value = context.portfolio.portfolio_value
        position_current_value = {pos.sid: pos.amount* pos.last_sale_price for i,pos in context.portfolio.positions.items()}
        
        # 买入
        if  context.trading_day_index % context.buy_frequency == 0:
            if len(stock_hold_now) >= context.max_stock_count:
                return 
            
            today_buy_count = 0
            if context.trade_mode == '轮动':
                for s in target_stock_to_buy:
                    if today_buy_count + len(stock_hold_now) >= context.max_stock_count: # 超出最大持仓数量
                        break
                    if data.can_trade(context.symbol(s)):
                        order_target_percent(context.symbol(s), equal_weight)
                        today_buy_count += 1
            else:
                if context.can_duplication_buy == True: # 可以重复买入,多一份买入
                    for s in target_stock_to_buy:
                        if today_buy_count + len(stock_hold_now) >= context.max_stock_count: # 超出最大持仓数量
                            break
                            
                        if data.can_trade(context.symbol(s)):
                            if context.symbol(s) in position_current_weight:
                                curr_value = position_current_value.get(context.symbol(s)) 
                                order_value(context.symbol(s), min(context.max_stock_weight * portfolio_value - curr_value, equal_weight*portfolio_value))
                            else:
                                order_value(context.symbol(s), equal_weight*portfolio_value)
                            today_buy_count += 1
    
                elif context.can_duplication_buy == False: # 不可以重复买入,不买
                    for s in target_stock_to_buy:
                        if today_buy_count + len(stock_hold_now) >= context.max_stock_count: # 超出最大持仓数量
                            break
                        if s in stock_hold_now:
                            continue
                        else:
                            if data.can_trade(context.symbol(s)):
                                order_target_percent(context.symbol(s), equal_weight)
                                today_buy_count += 1
    
                            
    #--------------------------------------------------------------------
    # 风控体系
    #--------------------------------------------------------------------                         
    def market_risk_manage(context, data):
        """大盘风控"""
        date = data.current_dt.strftime('%Y-%m-%d')
        if type(context.index_signal_data) == pd.DataFrame:
            current_signal = context.index_signal_data.loc[date]['signal']
            if current_signal == 'short': 
                stock_hold_now = [equity.symbol for equity in context.portfolio.positions]  
                # 平掉所有股票
                for stock in stock_hold_now:
                    if data.can_trade(context.symbol(stock)):
                        context.order_target_percent(context.symbol(stock), 0) 
                print('大盘出现止损信号, 平掉全部仓位,并关闭交易!')
                context.market_risk_signal = 'short'
        else:
            context.market_risk_signal = 'long'
    
            
            
    def strategy_risk_manage(context, data):
        """策略风控"""
        if context.strategy_risk_conf == []: # 没有设置策略风控
            context.strategy_risk_signal = 'long'
        
        else:
            for rm in context.strategy_risk_conf:
                if rm['method'] == 'strategy_percent_stopwin':
                    pct = rm['params']['percent']
                    portfolio_value = context.portfolio.portfolio_value 
                    if  portfolio_value / context.capital_base - 1 > pct: 
                        stock_hold_now = [equity.symbol for equity in context.portfolio.positions]  
                        # 平掉所有股票
                        for stock in stock_hold_now:
                            if data.can_trade(context.symbol(stock)):
                                context.order_target_percent(context.symbol(stock), 0) 
                        print('策略出现止盈信号, 平掉全部仓位,并关闭交易!')
                        context.strategy_risk_signal = 'short'  
                    
                
                if rm['method'] == 'strategy_percent_stoploss':
                    pct = rm['params']['percent']
                    portfolio_value = context.portfolio.portfolio_value 
                    if  portfolio_value / context.capital_base -1 < pct:
                        stock_hold_now = [equity.symbol for equity in context.portfolio.positions]  
                        # 平掉所有股票
                        for stock in stock_hold_now:
                            if data.can_trade(context.symbol(stock)):
                                context.order_target_percent(context.symbol(stock), 0) 
                        print('策略出现止损信号, 平掉全部仓位,并关闭交易!')
                        context.strategy_risk_signal = 'short'
    
            
    def stock_risk_manage(context, data):
        """个股风控"""
        position_current_pnl = {pos.sid: (pos.last_sale_price-pos.cost_basis)/pos.cost_basis for i,pos in context.portfolio.positions.items()}
        
        for rm in context.stock_risk_conf:
            params_pct = rm['params']['percent']
            if rm['method'] == 'stock_percent_stopwin':
                for sid,pnl_pct in position_current_pnl.items():  
                    if pnl_pct > params_pct:
                        context.stock_hit_stop.append(sid.symbol)
              
            if rm['method'] == 'stock_percent_stoploss':
                for sid,pnl_pct in position_current_pnl.items():
                    if pnl_pct < params_pct:
                        context.stock_hit_stop.append(sid.symbol)
    
    
    
    
    # 回测引擎:每日数据处理函数,每天执行一次
    def m4_handle_data_bigquant_run(context, data):
        """每日运行策略逻辑"""
        market_risk_manage(context, data)
        strategy_risk_manage(context, data)
        
        if context.market_risk_signal == 'short': return
        if context.strategy_risk_signal == 'short': return
    
        stock_risk_manage(context, data)
        
        sell_action(context, data)
        buy_action(context, data)
    
    # 回测引擎:准备数据,只执行一次
    def m4_prepare_bigquant_run(context):
          
        load_data = context.options['data'].read_pickle()
        context.signal_daily_stock = load_data['df1'].groupby('date').apply(lambda x:list(x.instrument))
        context.enter_daily_df = load_data['df2'].groupby('date').apply(lambda x:list(x.instrument))
        context.exit_daily_df = load_data['df3'].groupby('date').apply(lambda x:list(x.instrument))
    
    # 回测引擎:每个单位时间开始前调用一次,即每日开盘前调用一次。
    def m4_before_trading_start_bigquant_run(context, data):
        
        """每日盘前更新股票池"""
        frequency = context.rebalance_periods if context.trade_mode == '轮动' else context.stock_select_frequency
        if context.trading_day_index % frequency == 0:
            date = data.current_dt.strftime('%Y-%m-%d')
            try:
                context.selected_stock = context.signal_daily_stock[date] 
            except KeyError as e:
                context.selected_stock = []
        
        """初始化风控参数"""
        context.strategy_risk_signal = 'long'
        context.market_risk_signal = 'long' 
        context.stock_hit_stop = []
        context.cannot_buy_stock = []
    
    
    m1 = M.instruments.v2(
        start_date='2020-01-01',
        end_date='2022-12-16',
        market='CN_STOCK_A',
        instrument_list=''
    )
    
    m8 = M.cached.v3(
        input_1=m1.data,
        run=m8_run_bigquant_run,
        post_run=m8_post_run_bigquant_run,
        input_ports='input_1',
        params='{\'input_index\':\'000300.HIX\'}',
        output_ports='data_1'
    )
    
    m3 = M.input_features.v1(
        features="""
    in_csi300_0
    in_csi500_0
    in_sse50_0
    industry_sw_level1_0
    st_status_0
    
    # 选股条件
    cond1=(ta_kdj_j(high_0/adjust_factor_0, low_0/adjust_factor_0, close_0/adjust_factor_0, N=9, M1=3, M2=3)<-6)&\
    (fs_net_profit_yoy_0>1.5)&\
    (ta_macd(close_0/adjust_factor_0, fastperiod=6, slowperiod=38, signalperiod=6, derive='short'))
    
    # 排序选股
    cond2=1
    
    # 进场条件
    cond3=(ta_inverted_hammer(high_0/adjust_factor_0, low_0/adjust_factor_0, close_0/adjust_factor_0, open_0/adjust_factor_0))
                  
    # 卖出条件
    cond4=1
    """
    )
    
    m15 = M.general_feature_extractor.v7(
        instruments=m1.data,
        features=m3.data,
        start_date='',
        end_date='',
        before_start_days=300
    )
    
    m16 = M.derived_feature_extractor.v3(
        input_data=m15.data,
        features=m3.data,
        date_col='date',
        instrument_col='instrument',
        drop_na=False,
        remove_extra_columns=False
    )
    
    m10 = M.input_features.v1(
        features="""concept
    """
    )
    
    m5 = M.use_datasource.v1(
        instruments=m1.data,
        features=m10.data,
        datasource_id='industry_CN_STOCK_A',
        start_date='',
        end_date=''
    )
    
    m7 = M.join.v3(
        data1=m5.data,
        data2=m16.data,
        on='date,instrument',
        how='inner',
        sort=False
    )
    
    m6 = M.input_features.v1(
        features='suspended'
    )
    
    m19 = M.use_datasource.v1(
        instruments=m1.data,
        features=m6.data,
        datasource_id='stock_status_CN_STOCK_A',
        start_date='',
        end_date=''
    )
    
    m20 = M.join.v3(
        data1=m7.data,
        data2=m19.data,
        on='date,instrument',
        how='inner',
        sort=False
    )
    
    m22 = M.cached.v3(
        input_1=m20.data,
        run=m22_run_bigquant_run,
        post_run=m22_post_run_bigquant_run,
        input_ports='',
        params='{}',
        output_ports=''
    )
    
    m17 = M.cached.v3(
        input_1=m22.data_1,
        input_2=m22.data_2,
        input_3=m22.data_3,
        run=m17_run_bigquant_run,
        post_run=m17_post_run_bigquant_run,
        input_ports='',
        params='{}',
        output_ports=''
    )
    
    m4 = M.trade.v4(
        instruments=m1.data,
        options_data=m17.data_1,
        benchmark_ds=m8.data_1,
        start_date='',
        end_date='',
        initialize=m4_initialize_bigquant_run,
        handle_data=m4_handle_data_bigquant_run,
        prepare=m4_prepare_bigquant_run,
        before_trading_start=m4_before_trading_start_bigquant_run,
        volume_limit=0.025,
        order_price_field_buy='open',
        order_price_field_sell='open',
        capital_base=30000,
        auto_cancel_non_tradable_orders=True,
        data_frequency='daily',
        price_type='真实价格',
        product_type='股票',
        plot_charts=True,
        backtest_only=False,
        benchmark='000300.SHA'
    )
    
    ---------------------------------------------------------------------------
    KeyError                                  Traceback (most recent call last)
    <ipython-input-1-35d0cf4ac64b> in <module>
        527 )
        528 
    --> 529 m4 = M.trade.v4(
        530     instruments=m1.data,
        531     options_data=m17.data_1,
    
    <ipython-input-1-35d0cf4ac64b> in m4_handle_data_bigquant_run(context, data)
        379     stock_risk_manage(context, data)
        380 
    --> 381     sell_action(context, data)
        382     buy_action(context, data)
        383 
    
    <ipython-input-1-35d0cf4ac64b> in sell_action(context, data)
        226             if data.can_trade(context.symbol(stock)):
        227                 context.order_target_percent(context.symbol(stock), 0)
    --> 228                 del context.portfolio.positions[context.symbol(stock)]
        229 
        230 
    
    KeyError: Equity(4419 [002891.SZA])