liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-17 01:14:35 +08:00
Code Activity

simplify the code and prevent float when shifting

Browse Source
This commit is contained in:
Young
2021-06-20 08:03:00 +00:00
committed by you-n-g
parent 0aee46ee79
commit a3679e6758
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/data/cache.py
View File

@@ -784,10 +784,10 @@ class DiskDatasetCache(DatasetCache):
def build_index_from_data(data, start_index=0): def build_index_from_data(data, start_index=0):
if data.empty: if data.empty:
return pd.DataFrame() return pd.DataFrame()
line_data = data.iloc[:, 0].fillna(0).groupby("datetime").count() line_data = data.groupby("datetime").size()
line_data.sort_index(inplace=True) line_data.sort_index(inplace=True)
index_end = line_data.cumsum() index_end = line_data.cumsum()
index_start = index_end.shift(1).fillna(0) index_start = index_end.shift(1, fill_value=0)
index_data = pd.DataFrame() index_data = pd.DataFrame()
index_data["start"] = index_start index_data["start"] = index_start