Skip to content

A
alpha-mind
Commit 41346f12 authored by Dr.李's avatar Dr.李
Browse files
Options

added back trade_date

parent 14ea7d7c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 6 deletions
alphamind/data/engines/sqlengine.py
View file @ 41346f12
...@@ -554,9 +554,7 @@ class SqlEngine(object): ...@@ -554,9 +554,7 @@ class SqlEngine(object):
out_s.append(i) out_s.append(i)
res = df[['code', 'industry_code'] + in_s] res = df[['code', 'industry_code'] + in_s]
res = res.assign(**dict(zip(out_s, [0] * len(out_s))))
for i in out_s:
res[i] = 0
return res return res
def fetch_industry_range(self, def fetch_industry_range(self,
...@@ -611,10 +609,9 @@ class SqlEngine(object): ...@@ -611,10 +609,9 @@ class SqlEngine(object):
else: else:
out_s.append(i) out_s.append(i)
res = df[['code', 'industry_code'] + in_s] res = df[['trade_date', 'code', 'industry_code'] + in_s]
for i in out_s: res = res.assign(**dict(zip(out_s, [0]*len(out_s))))
res[i] = 0
return res return res
def fetch_data(self, ref_date: str, def fetch_data(self, ref_date: str,
......
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