liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-10 06:20:57 +08:00
Code Activity

fix SepDataFrame when we del it to empty (#1082)

Browse Source
This commit is contained in:
you-n-g
2022-04-29 14:29:17 +08:00
committed by GitHub
parent 9d0a8f61d1
commit 3c9c76b384
2 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
qlib/contrib/data/utils/sepdf.py
View File

@@ -24,6 +24,10 @@ class SepDataFrame:
SepDataFrame tries to act like a DataFrame whose column with multiindex
"""
# TODO:
# SepDataFrame try to behave like pandas dataframe, but it is still not them same
# Contributions are welcome to make it more complete.
def __init__(self, df_dict: Dict[str, pd.DataFrame], join: str, skip_align=False):
"""
initialize the data based on the dataframe dictionary
@@ -77,7 +81,11 @@ class SepDataFrame:
def _update_join(self):
if self.join not in self:
self.join = next(iter(self._df_dict.keys()))
if len(self._df_dict) > 0:
self.join = next(iter(self._df_dict.keys()))
else:
# NOTE: this will change the behavior of previous reindex when all the keys are empty
self.join = None
def __getitem__(self, item):
# TODO: behave more like pandas when multiindex