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

Added typehints for head and tail

Browse Source
This commit is contained in:
Gourav Kumar 2022-03-29 10:35:20 +05:30
parent 3b96d231f4
commit 875089349e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
fincal/core.py
View File

@ -1,3 +1,5 @@
from __future__ import annotations
import datetime
import inspect
from collections import UserDict, UserList
@ -423,12 +425,12 @@ class TimeSeriesCore(UserDict):
return _IndexSlicer(self)
def head(self, n: int = 6):
def head(self, n: int = 6) -> TimeSeriesCore:
"""Returns the first n items of the TimeSeries object"""
return self.iloc[:n]
def tail(self, n: int = 6):
def tail(self, n: int = 6) -> TimeSeriesCore:
"""Returns the last n items of the TimeSeries object"""
return self.iloc[-n:]