gourav/PyFacts
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

improved slicer class

Browse Source
This commit is contained in:
Gourav Kumar 2022-02-21 23:07:43 +05:30
parent 03283f7ed4
commit 04176646b6
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
fincal/core.py
View File

@ -115,16 +115,17 @@ def _parse_date(date: str, date_format: str = None):
class _IndexSlicer:
"""Class to create a slice using iloc in TimeSeriesCore"""
def __init__(self, parent_obj):
self.parent = parent_obj
def __getitem__(self, n):
all_keys = list(self.parent.time_series)
if isinstance(n, int):
keys = [all_keys[n]]
keys = [self.parent.dates[n]]
else:
keys = all_keys[n]
item = [(key, self.parent.time_series[key]) for key in keys]
keys = self.parent.dates[n]
item = [(key, self.parent.data[key]) for key in keys]
if len(item) == 1:
return item[0]