复制代码
stg_res = pd.DataFrame()
stg_res.loc[0, '选币因子'] = str(factor_list)
stg_res.loc[0, '过滤因子'] = str(filter_list)
stg_res.loc[0, '持仓周期'] = hold_period
stg_res.loc[0, 'offset'] = offset
stg_res.loc[0, '选币数量'] = select_factor_num_str
stg_res.loc[0, '是否使用现货'] = if_use_spot
year_return['涨跌幅'] = year_return['涨跌幅'].map(lambda x: float(x[:-1]))
stg_res['各年收益'] = [year_return.values.reshape(-1).tolist()]
# 记录回测的各项指标
cols = ['累积净值', '年化收益', '最大回撤']
long_rename_cols = {'累积净值': '多头累积净值', '年化收益': '多头年化收益', '最大回撤': '多头最大回撤'}
long_rename_cols1 = {'累积净值': '多头累积净值_择时', '年化收益': '多头年化收益_择时', '最大回撤': '多头最大回撤_择时'}
short_rename_cols = {'累积净值': '空头累积净值', '年化收益': '空头年化收益', '最大回撤': '空头最大回撤'}
short_rename_cols1 = {'累积净值': '空头累积净值_择时', '年化收益': '空头年化收益_择时', '最大回撤': '空头最大回撤_择时'}
short_rename_cols2 = {'累积净值': '空头累积净值_定投', '年化收益': '空头年化收益_定投', '最大回撤': '空头最大回撤_定投'}
long_rtn = long_rtn.T
long_rtn = long_rtn[cols]
long_rtn = long_rtn.rename(columns=long_rename_cols)
short_rtn = short_rtn.T
short_rtn = short_rtn[cols]
short_rtn = short_rtn.rename(columns=short_rename_cols)
# 记录回测的各项指标
stg_res = pd.concat([stg_res, rtn.T, long_rtn, short_rtn], axis=1)
# 保存文件
if os.path.exists(result_path): # 如果文件存在，往原有的文件中添加新的结果
stg_res.to_csv(result_path, encoding='gbk', index=False, header=False, mode='a')
else: # 如果文不件存在，常规的to_csv操作
stg_res.to_csv(result_path, encoding='gbk', index=False)