mayutils 1.2.30__tar.gz → 1.2.32__tar.gz

{mayutils-1.2.30 → mayutils-1.2.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mayutils
-Version: 1.2.30
+Version: 1.2.32
 Summary: Utilities for Python from Mayuran Visakan
 Author: Mayuran Visakan
 Author-email: Mayuran Visakan <mayuran.k.v@gmail.com>

{mayutils-1.2.30 → mayutils-1.2.32}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mayutils"
-version = "1.2.30"
+version = "1.2.32"
 description = "Utilities for Python from Mayuran Visakan"
 authors = [
     { name = "Mayuran Visakan", email = "mayuran.k.v@gmail.com" }

{mayutils-1.2.30 → mayutils-1.2.32}/src/mayutils/objects/dataframes.py

@@ -27,6 +27,7 @@ from dataframe_image._pandas_accessor import (
 )
 import polars as pl
 import numpy as np
+from mayutils.objects.datetime import Interval
 from mayutils.objects.colours import Colour
 from mayutils.export import OUTPUT_FOLDER
 
@@ -411,6 +412,25 @@ class DataframeUtilsAccessor(object):
 
         return self.df
 
+    def ground(
+        self,
+        interval: Optional[Interval] = None,
+    ) -> DataFrame:
+        if interval is None:
+            return self.df
+
+        raise NotImplementedError("Grounding not implemented for DataFrames yet")
+        # if self.df.index.inferred_type == "datetime":
+        #     grounding_value = self.df.loc[interval.as_slice].mean()
+        # elif self.df.index.inferred_type == "date":
+        #     grounding_value = self.df.loc[interval.as_date_slice].mean()
+        # else:
+        #     raise TypeError("Series index must be datetime or date type for grounding")
+
+        # grounded_series = self.series.div(grounding_value)
+
+        # return grounded_series
+
 
 class SeriesUtilsAccessor(object):
     def __init__(
@@ -428,6 +448,24 @@ class SeriesUtilsAccessor(object):
             "Not implemented for series yet: leverage existing df methods"
         )
 
+    def ground(
+        self,
+        interval: Optional[Interval] = None,
+    ) -> Series:
+        if interval is None:
+            return self.series
+
+        if self.series.index.inferred_type == "datetime":
+            grounding_value = self.series.loc[interval.as_slice].mean()
+        elif self.series.index.inferred_type == "date":
+            grounding_value = self.series.loc[interval.as_date_slice].mean()
+        else:
+            raise TypeError("Series index must be datetime or date type for grounding")
+
+        grounded_series = self.series.div(grounding_value)
+
+        return grounded_series
+
 
 class IndexUtilsAccessor(object):
     def __init__(

{mayutils-1.2.30 → mayutils-1.2.32}/src/mayutils/visualisation/graphs/plotly/charts.py

@@ -14,6 +14,7 @@ from plotly._subplots import _build_subplot_title_annotations
 
 from mayutils.objects.colours import TRANSPARENT, Colour
 
+from mayutils.objects.datetime import Interval
 from mayutils.objects.functions import set_inline
 
 from mayutils.visualisation.graphs.plotly.templates import (
@@ -1276,6 +1277,24 @@ class Plot(go.Figure):
 
         return self
 
+    def add_interval(
+        self,
+        interval: Optional[Interval],
+        **kwargs,
+    ) -> Self:
+        if interval is None:
+            return self
+
+        kwargs = {"line_width": 0, "opacity": 0.1} | kwargs
+
+        self.add_vrect(
+            x0=interval.start.simple,
+            x1=interval.end.simple,
+            **kwargs,
+        )
+
+        return self
+
     def __call__(
         self,
         save: bool = True,