Skip to content

R
robo-dividend
Commit 3cc9723a authored by 吕先亚's avatar 吕先亚
Browse files
Options

bugfix

parent 9a60da9c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
basic/sp500.py
View file @ 3cc9723a
...@@ -170,8 +170,9 @@ def sync_sp500(day): ...@@ -170,8 +170,9 @@ def sync_sp500(day):
datas.append(data) datas.append(data)
wb.close() wb.close()
datas = pd.DataFrame(datas[::-1]) datas = pd.DataFrame(datas[::-1])
datas['eps'] = datas['eps'].shift(-1).ffill() datas['eps'] = datas['eps'].shift(-1)
datas['close'] = datas['eps'].rolling(window=4).sum().round(2) datas['close'] = datas['eps'].rolling(window=4).sum().round(2)
datas.ffill(inplace=True)
datas.dropna(inplace=True) datas.dropna(inplace=True)
return datas.to_dict(orient="records")[-1::] if day else datas.to_dict(orient="records") return datas.to_dict(orient="records")[-1::] if day else datas.to_dict(orient="records")
......
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