Commit ecfe86a2 authored by Dr.李's avatar Dr.李

added shortname

parent b81fc005
......@@ -359,6 +359,7 @@ class SqlEngine(object):
res = transformer.transform('code', df).replace([-np.inf, np.inf], np.nan)
res['chgPct'] = df.chgPct
res['secShortName'] = df['secShortName']
res = res.loc[ref_date]
res.index = list(range(len(res)))
return res
......@@ -421,6 +422,7 @@ class SqlEngine(object):
res = transformer.transform('code', df).replace([-np.inf, np.inf], np.nan)
res['chgPct'] = df.chgPct
res['secShortName'] = df['secShortName']
res = res.reset_index()
return pd.merge(res, universe_df[['trade_date', 'code']], how='inner').drop_duplicates(['trade_date', 'code'])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment