沪深股票的复权计算(复权因子的应用)--代码实现


对复权因子的介绍可参考

通过计算复权因子求得当日前一天收盘价,并存储在MongoDB数据库中
代码实现

def fill_au_factor_pre_close(begin_date,end_date): """ 为daily数据集填充 1、复权因子au_factor,复权因子计算方式:au_factor= hfq_close/close 2、pre_close = close(-1) * au_factor(-1)/au_factor :param begin_date:开始日期 :param end_date:结束日期 """ all_codes = get_all_codes() for code in all_codes: hfq_daily_cursor = DB_CONN['daily_hfq'].find( {'code':code,'date':{'$gte':begin_date,'$lte':end_date},'index':False}, sort=[('date',ASCENDING)], projection={'date':True,'close':True} ) date_hfq_close_dict = dict([(x['date'],x['close']) for x in hfq_daily_cursor]) daily_cursor = DB_CONN['daily'].find( {'code':code,'date':{'$gte':begin_date,'$lte':end_date},'index':False}, sort=[('date',ASCENDING)], projection={'date':True,'close':True} ) last_close = -1 last_au_factor = -1 update_requests = [] for daily in daily_cursor: date = daily['date'] try: close = daily['close'] doc = dict() #复权因子 = 当日后复权价格/当日的实际价格 au_factor = round(date_hfq_close_dict['date']/close,2) doc['au_factor'] = au_factor #当日前收价=前一日实际收盘价 * 前一日复权因子/当日复权因子 if last_au_factor != -1 and last_close != -1: pre_close = last_close * last_au_factor / au_factor doc['pre_close'] = round(pre_close,2) last_close = close last_au_factor = au_factor update_requests.append( UpdateOne( {'code':code,'date':date,'index':False}, {'$set':doc} ) ) except: print('计算复权因子时发生错误,股票代码:%s,日期:%s' %(code,date),flush=True) #恢复成初始值,防止出错 last_close = -1 last_au_factor = -1 if len(update_requests)>0: update_result = DB_CONN['daily'].bulk_write(update_requests,ordered=False) print('填充复权因子和前收,股票:%s ,更新:%4d条'%(code,update_result.modified_count),flush=False) 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM