import pandas as pd import numpy as np df_day = pd.read_csv('000002_daily.csv',encoding = 'gbk') df_zjc = pd.read_csv('000002_zjc.csv',encoding = 'gbk') df_day = df_day[['ts_code', 'trade_date', 'close']] df_day['date'] = pd.to_datetime(df_day['trade_date'],format='%Y%m%d') df_day = df_day.set_index('date',drop = True) df_zjc=df_zjc[['ts_code', 'ann_date']] df_zjc['date'] = pd.to_datetime(df_zjc['ann_date'],format='%Y%m%d') df_zjc = df_zjc.set_index('date',drop = True) for i in range(0,len(df_zjc)): df_zjc.loc[df_zjc.index[i],'1day'] = df_day['close'][df_zjc.index[i]:][1] df_zjc.loc[df_zjc.index[i],'5day'] = df_day['close'][df_zjc.index[i]:][4] df_zjc.loc[df_zjc.index[i],'15day'] = df_day['close'][df_zjc.index[i]:][14]