MatplotLibAPI 3.0.4__py3-none-any.whl → 3.0.5__py3-none-any.whl

MatplotLibAPI/Bubble.py

@@ -15,9 +15,7 @@ BUBBLE_STYLE_TEMPLATE = StyleTemplate(
                   "y": percent_formatter,
                   "label": string_formatter,
                   "z": bmk_formatter},
-    yscale="log",
-    y_ticks=8,
-    x_ticks=8
+    yscale="log"
 )
 
 

MatplotLibAPI/Composite.py

@@ -6,7 +6,7 @@ import pandas as pd
 from .Bubble import plot_bubble, BUBBLE_STYLE_TEMPLATE
 from .Table import plot_table
 from typing import Optional, Tuple
-from .Style import StyleTemplate, _validate_panda,format_func
+from .Style import StyleTemplate, _validate_panda, format_func
 
 
 def plot_composite_bubble(
@@ -19,18 +19,27 @@ def plot_composite_bubble(
         style: StyleTemplate = BUBBLE_STYLE_TEMPLATE,
         max_values: int = 50,
         center_to_mean: bool = False,
+        filter_by:Optional[str] = None,
         sort_by: Optional[str] = None,
         ascending: bool = False,
         table_rows: int = 10,
         figsize: Tuple[float, float] = (19.2, 10.8)) -> Figure:
 
     _validate_panda(pd_df, cols=[label, x, y, z], sort_by=sort_by)
-    style.format_funcs=format_func(style.format_funcs,label=label,x=x,y=y)
+
+    if not sort_by:
+        sort_by = z
+    if not filter_by:
+        filter_by = z
+    plot_df = pd_df.sort_values(by=filter_by,
+                                ascending=ascending)[[label, x, y, z]].head(max_values)
+    style.format_funcs = format_func(
+        style.format_funcs, label=label, x=x, y=y, z=z)
     fig = plt.figure(figsize=figsize)
     fig.patch.set_facecolor("black")
     grid = plt.GridSpec(2, 2, height_ratios=[2, 1], width_ratios=[1, 1])
     ax = fig.add_subplot(grid[0, 0:])
-    ax = plot_bubble(pd_df=pd_df,
+    ax = plot_bubble(pd_df=plot_df,
                      label=label,
                      x=x,
                      y=y,
@@ -54,7 +63,7 @@ def plot_composite_bubble(
 
     ax2 = fig.add_subplot(grid[1, 0])
     ax2 = plot_table(
-        pd_df=pd_df,
+        pd_df=plot_df,
         cols=[label, z, y, x],
         title=f"Top {table_rows}",
         ax=ax2,
@@ -65,7 +74,7 @@ def plot_composite_bubble(
     )
     ax3 = fig.add_subplot(grid[1, 1])
     ax3 = plot_table(
-        pd_df=pd_df,
+        pd_df=plot_df,
         cols=[label, z, y, x],
         title=f"Worst {table_rows}",
         ax=ax3,

MatplotLibAPI/Treemap.py

@@ -0,0 +1,59 @@
+from typing import List, Optional
+import pandas as pd
+import plotly.graph_objects as go
+from .Style import DynamicFuncFormatter, StyleTemplate, generate_ticks, string_formatter, _validate_panda, percent_formatter, format_func
+
+TREEMAP_STYLE_TEMPLATE = StyleTemplate(
+    background_color='black',
+    fig_border='darkgrey',
+    font_color='white',
+    palette='magma',
+    format_funcs={"y": percent_formatter,
+                  "label": string_formatter}
+)
+
+
+def plot_treemap(pd_df: pd.DataFrame,
+                 path: str,
+                 values: str,
+                 style: StyleTemplate = TREEMAP_STYLE_TEMPLATE,
+                 title: Optional[str] = None,
+                 color: Optional[str] = None,
+                 sort_by: Optional[str] = None,
+                 ascending: bool = False,
+                 max_values: int = 100,
+                 fig: Optional[go.Figure] = None) -> go.Figure:
+    cols = [path, values]
+    if color:
+        cols.append(color)
+    _validate_panda(df, cols=cols, sort_by=sort_by)
+    if not sort_by:
+        sort_by = values
+    df = pd_df.sort_values(by=sort_by, ascending=ascending)[
+        cols].head(max_values)
+    data = {"labels": df[path],
+            "parents": [""] * len(df),
+            "values": df[values],
+            "textinfo": "label",
+            "name": title}
+
+    if color:
+        df['color'] = df[color]
+
+    if not fig:
+        fig = go.Figure(data=data)
+    else:
+        fig.add_trace(go.Treemap(data))
+    
+    fig.update_layout(
+        paper_bgcolor=style.background_color,
+        plot_bgcolor=style.background_color,
+        font=dict(color=style.font_color),
+        margin=dict(t=50, l=25, r=25, b=25))
+
+    # Apply color scale
+    fig.update_traces(
+        marker=dict(colorscale=style.palette)
+    )
+
+    return fig

MatplotLibAPI/__init__.py

@@ -5,9 +5,10 @@ from .Timeserie import plot_timeserie
 from .Bubble import plot_bubble
 from .Network import plot_network
 from .Pivot import plot_pivotbar
+from .Treemap import plot_treemap
 from .Composite import plot_composite_bubble
 from .pdAccessor import MatPlotLibAccessor
 from .Style import StyleTemplate
 
 __all__ = ["plot_bubble", "plot_timeserie", "plot_table", "plot_network",
-           "plot_pivotbar", "plot_composite_bubble", "StyleTemplate", "MatPlotLibAccessor"]
+           "plot_pivotbar", "plot_treemap","plot_composite_bubble", "StyleTemplate", "MatPlotLibAccessor"]

MatplotLibAPI/pdAccessor.py

@@ -1,17 +1,20 @@
 
 import logging
 import warnings
+
 from typing import Optional, List
+import pandas as pd
+
 from matplotlib.axes import Axes
 from matplotlib.figure import Figure
-import pandas as pd
+import plotly.graph_objects as go
 from .Style import StyleTemplate
 from .Bubble import plot_bubble, BUBBLE_STYLE_TEMPLATE
 from .Composite import plot_composite_bubble
 from .Timeserie import plot_timeserie, TIMESERIE_STYLE_TEMPLATE
 from .Table import plot_table, TABLE_STYLE_TEMPLATE
-from .Network import (Graph)
-
+from .Network import Graph
+from .Treemap import plot_treemap, TREEMAP_STYLE_TEMPLATE
 
 warnings.filterwarnings('ignore')
 logging.getLogger().setLevel(logging.WARNING)
@@ -28,7 +31,7 @@ class MatPlotLibAccessor:
                     x: str,
                     y: str,
                     z: str,
-                    title: str = "Test",
+                    title: Optional[str] = None,
                     style: StyleTemplate = BUBBLE_STYLE_TEMPLATE,
                     max_values: int = 50,
                     center_to_mean: bool = False,
@@ -47,14 +50,14 @@ class MatPlotLibAccessor:
                            sort_by=sort_by,
                            ascending=ascending)
 
-    def plot_bubble_composite(self,
+    def plot_composite_bubble(self,
                               label: str,
                               x: str,
                               y: str,
                               z: str,
                               title: Optional[str] = None,
                               style: StyleTemplate = BUBBLE_STYLE_TEMPLATE,
-                              max_values: int = 50,
+                              max_values: int = 100,
                               center_to_mean: bool = False,
                               sort_by: Optional[str] = None,
                               ascending: bool = False) -> Figure:
@@ -93,7 +96,7 @@ class MatPlotLibAccessor:
                        y: str,
                        title: Optional[str] = None,
                        style: StyleTemplate = TIMESERIE_STYLE_TEMPLATE,
-                       max_values: int = 20,
+                       max_values: int = 100,
                        sort_by: Optional[str] = None,
                        ascending: bool = False) -> Axes:
 
@@ -123,3 +126,22 @@ class MatPlotLibAccessor:
                                            weight=weight)
 
         return graph.plotX(title, style)
+    
+    def plot_treemap(self,
+                 path: str,
+                 values: str,
+                 style: StyleTemplate = TREEMAP_STYLE_TEMPLATE,
+                 title: Optional[str] = None,
+                 color: Optional[str] = None, 
+                 max_values: int = 100,
+                 sort_by: Optional[str] = None,
+                 ascending: bool = False) ->go.Figure:
+        return plot_treemap(pd_df=self._obj,
+                              path=path,
+                              values=values,
+                              title=title,
+                              style=style,
+                              color=color,
+                              max_values=max_values,
+                              sort_by=sort_by,
+                              ascending=ascending)

{MatplotLibAPI-3.0.4.dist-info → MatplotLibAPI-3.0.5.dist-info}/METADATA

@@ -1,12 +1,13 @@
 Metadata-Version: 2.1
 Name: MatplotLibAPI
-Version: 3.0.4
+Version: 3.0.5
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pandas
 Requires-Dist: matplotlib
 Requires-Dist: networkx
+Requires-Dist: plotly
 Requires-Dist: seaborn
 Requires-Dist: scikit-learn
 

MatplotLibAPI-3.0.5.dist-info/RECORD

@@ -0,0 +1,15 @@
+MatplotLibAPI/Bubble.py,sha256=2PIwL0stzFC7knqdMHBBm99DXPXtBJK-xmpxmVlInYw,3830
+MatplotLibAPI/Composite.py,sha256=M_AaBdn9cfMIOGZBg4_twRS9VU9jbgS6cs9E5l3ObOQ,2763
+MatplotLibAPI/Network.py,sha256=fq1-LfDKy3oC5JzqU7IyZudgR2plmziigjpm2TzEA3U,13286
+MatplotLibAPI/Pivot.py,sha256=8jRYdlvw8otrJ1hYBAaKQJoN84xe-YfroXmgRLykMmQ,7162
+MatplotLibAPI/Style.py,sha256=xtw71VWHSQEvts1nJdKR0c_9Rqef6EGkyyvgdTK5M7c,4632
+MatplotLibAPI/Table.py,sha256=Az5uX-ViqzJPTBym4QnUzn5PD5LEsVzQ9WohcSDFcK4,1990
+MatplotLibAPI/Timeserie.py,sha256=HRFHs_WzaZzLmWVVMBKVMP5M3oglsr1R9Ni53X7_HLI,3432
+MatplotLibAPI/Treemap.py,sha256=diCx5270ENBObO3eWlLlpaHWUWcSgMT34vEwWTxTDAE,1799
+MatplotLibAPI/__init__.py,sha256=PDobJg0sw7BjgSQk7p7kkjombghs3D_m-CCk2oeUt_s,506
+MatplotLibAPI/pdAccessor.py,sha256=5duWsURQKgNuwCz1p0ge2GbAXHSHl8e38zJn6NSVAGg,5745
+MatplotLibAPI-3.0.5.dist-info/LICENSE,sha256=hMErKLb6YZR3lRR5zr-vxeFkvY69QAaafgSpZ5-P1dQ,1067
+MatplotLibAPI-3.0.5.dist-info/METADATA,sha256=icSnJdKd4E6mw4SKUPaTHg9QgsWfUiiRvMlrQUXAz10,319
+MatplotLibAPI-3.0.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+MatplotLibAPI-3.0.5.dist-info/top_level.txt,sha256=MrzbBjDEW48Vb6YhQIqpFYGOhHzQnEIM5Qy2xy2iqew,14
+MatplotLibAPI-3.0.5.dist-info/RECORD,,

MatplotLibAPI-3.0.4.dist-info/RECORD

@@ -1,14 +0,0 @@
-MatplotLibAPI/Bubble.py,sha256=aYw8qcYF_cXJzAU7tIi5Cn7Y-sRfR8Cvm5RivMvVd5E,3860
-MatplotLibAPI/Composite.py,sha256=1qP_9TEz58EDA-oEev21WfJh2mxAEVVkRqYsysvMw7E,2478
-MatplotLibAPI/Network.py,sha256=fq1-LfDKy3oC5JzqU7IyZudgR2plmziigjpm2TzEA3U,13286
-MatplotLibAPI/Pivot.py,sha256=8jRYdlvw8otrJ1hYBAaKQJoN84xe-YfroXmgRLykMmQ,7162
-MatplotLibAPI/Style.py,sha256=xtw71VWHSQEvts1nJdKR0c_9Rqef6EGkyyvgdTK5M7c,4632
-MatplotLibAPI/Table.py,sha256=Az5uX-ViqzJPTBym4QnUzn5PD5LEsVzQ9WohcSDFcK4,1990
-MatplotLibAPI/Timeserie.py,sha256=HRFHs_WzaZzLmWVVMBKVMP5M3oglsr1R9Ni53X7_HLI,3432
-MatplotLibAPI/__init__.py,sha256=-i49Ta56gER88f6R_BDOo-JiLc5UUGvxHg2aHdJv0PQ,457
-MatplotLibAPI/pdAccessor.py,sha256=7R4QoZDHFkf67rn1zDxJMkghQa-BUPu6Fy7TXVT9-OQ,4844
-MatplotLibAPI-3.0.4.dist-info/LICENSE,sha256=hMErKLb6YZR3lRR5zr-vxeFkvY69QAaafgSpZ5-P1dQ,1067
-MatplotLibAPI-3.0.4.dist-info/METADATA,sha256=spXzV0K_jdy4D5-4oi4plFlFaltrVQRjtsJUXbbCH90,297
-MatplotLibAPI-3.0.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-MatplotLibAPI-3.0.4.dist-info/top_level.txt,sha256=MrzbBjDEW48Vb6YhQIqpFYGOhHzQnEIM5Qy2xy2iqew,14
-MatplotLibAPI-3.0.4.dist-info/RECORD,,