MatplotLibAPI 3.2.13__py3-none-any.whl → 3.2.15__py3-none-any.whl

MatplotLibAPI/Area.py

@@ -0,0 +1,76 @@
+"""Area chart helpers."""
+
+from typing import Any, Optional, Tuple
+
+import pandas as pd
+from matplotlib.axes import Axes
+from matplotlib.figure import Figure
+
+from .StyleTemplate import (
+    AREA_STYLE_TEMPLATE,
+    StyleTemplate,
+    string_formatter,
+    validate_dataframe,
+)
+from ._visualization_utils import _get_axis, _wrap_aplot
+
+
+def aplot_area(
+    pd_df: pd.DataFrame,
+    x: str,
+    y: str,
+    label: Optional[str] = None,
+    stacked: bool = True,
+    title: Optional[str] = None,
+    style: StyleTemplate = AREA_STYLE_TEMPLATE,
+    ax: Optional[Axes] = None,
+    **kwargs: Any,
+) -> Axes:
+    """Plot area charts, optionally stacked for part-to-whole trends."""
+    cols = [x, y]
+    if label:
+        cols.append(label)
+    validate_dataframe(pd_df, cols=cols)
+    plot_ax = _get_axis(ax)
+
+    if label:
+        pivot_df = pd_df.pivot_table(
+            index=x, columns=label, values=y, aggfunc="sum"
+        ).sort_index()
+        pivot_df.plot(kind="area", stacked=stacked, alpha=0.7, ax=plot_ax)
+    else:
+        sorted_df = pd_df.sort_values(by=x)
+        plot_ax.fill_between(
+            sorted_df[x], sorted_df[y], color=style.font_color, alpha=0.4
+        )
+        plot_ax.plot(sorted_df[x], sorted_df[y], color=style.font_color)
+
+    plot_ax.set_xlabel(string_formatter(x))
+    plot_ax.set_ylabel(string_formatter(y))
+    if title:
+        plot_ax.set_title(title)
+    return plot_ax
+
+
+def fplot_area(
+    pd_df: pd.DataFrame,
+    x: str,
+    y: str,
+    label: Optional[str] = None,
+    stacked: bool = True,
+    title: Optional[str] = None,
+    style: StyleTemplate = AREA_STYLE_TEMPLATE,
+    figsize: Tuple[float, float] = (10, 6),
+) -> Figure:
+    """Plot area charts on a new figure."""
+    return _wrap_aplot(
+        aplot_area,
+        pd_df=pd_df,
+        figsize=figsize,
+        x=x,
+        y=y,
+        label=label,
+        stacked=stacked,
+        title=title,
+        style=style,
+    )

MatplotLibAPI/Bar.py

@@ -0,0 +1,79 @@
+"""Bar and stacked bar chart helpers."""
+
+from typing import Any, Optional, Tuple
+
+import pandas as pd
+import seaborn as sns
+from matplotlib.axes import Axes
+from matplotlib.figure import Figure
+
+from .StyleTemplate import (
+    DISTRIBUTION_STYLE_TEMPLATE,
+    StyleTemplate,
+    string_formatter,
+    validate_dataframe,
+)
+from ._visualization_utils import _get_axis, _wrap_aplot
+
+
+def aplot_bar(
+    pd_df: pd.DataFrame,
+    category: str,
+    value: str,
+    group: Optional[str] = None,
+    stacked: bool = False,
+    title: Optional[str] = None,
+    style: StyleTemplate = DISTRIBUTION_STYLE_TEMPLATE,
+    ax: Optional[Axes] = None,
+    **kwargs: Any,
+) -> Axes:
+    """Plot bar or stacked bar charts for categorical comparisons."""
+    cols = [category, value]
+    if group:
+        cols.append(group)
+    validate_dataframe(pd_df, cols=cols)
+
+    plot_ax = _get_axis(ax)
+    plot_df = pd_df.copy()
+
+    if group:
+        pivot_df = plot_df.pivot_table(
+            index=category, columns=group, values=value, aggfunc="sum"
+        )
+        pivot_df.plot(kind="bar", stacked=stacked, ax=plot_ax, alpha=0.85)
+    else:
+        sns.barplot(
+            data=plot_df, x=category, y=value, palette=style.palette, ax=plot_ax
+        )
+
+    plot_ax.set_facecolor(style.background_color)
+    plot_ax.set_xlabel(string_formatter(category))
+    plot_ax.set_ylabel(string_formatter(value))
+    if title:
+        plot_ax.set_title(title)
+    plot_ax.tick_params(axis="x", labelrotation=45)
+    return plot_ax
+
+
+def fplot_bar(
+    pd_df: pd.DataFrame,
+    category: str,
+    value: str,
+    group: Optional[str] = None,
+    stacked: bool = False,
+    title: Optional[str] = None,
+    style: StyleTemplate = DISTRIBUTION_STYLE_TEMPLATE,
+    figsize: Tuple[float, float] = (10, 6),
+) -> Figure:
+    """Plot bar or stacked bar charts on a new figure."""
+    return _wrap_aplot(
+        aplot_bar,
+        pd_df=pd_df,
+        figsize=figsize,
+        category=category,
+        value=value,
+        group=group,
+        stacked=stacked,
+        title=title,
+        style=style,
+    )

MatplotLibAPI/BoxViolin.py

@@ -0,0 +1,69 @@
+"""Box and violin plot helpers."""
+
+from typing import Any, Optional, Tuple
+
+import pandas as pd
+import seaborn as sns
+from matplotlib.axes import Axes
+from matplotlib.figure import Figure
+
+from .StyleTemplate import (
+    DISTRIBUTION_STYLE_TEMPLATE,
+    StyleTemplate,
+    string_formatter,
+    validate_dataframe,
+)
+from ._visualization_utils import _get_axis, _wrap_aplot
+
+
+def aplot_box_violin(
+    pd_df: pd.DataFrame,
+    column: str,
+    by: Optional[str] = None,
+    violin: bool = False,
+    title: Optional[str] = None,
+    style: StyleTemplate = DISTRIBUTION_STYLE_TEMPLATE,
+    ax: Optional[Axes] = None,
+    **kwargs: Any,
+) -> Axes:
+    """Plot box or violin charts to summarize distributions."""
+    cols = [column]
+    if by:
+        cols.append(by)
+    validate_dataframe(pd_df, cols=cols)
+    plot_ax = _get_axis(ax)
+
+    if violin:
+        sns.violinplot(data=pd_df, x=by, y=column, palette=style.palette, ax=plot_ax)
+    else:
+        sns.boxplot(data=pd_df, x=by, y=column, palette=style.palette, ax=plot_ax)
+
+    plot_ax.set_facecolor(style.background_color)
+    plot_ax.set_ylabel(string_formatter(column))
+    if by:
+        plot_ax.set_xlabel(string_formatter(by))
+    if title:
+        plot_ax.set_title(title)
+    return plot_ax
+
+
+def fplot_box_violin(
+    pd_df: pd.DataFrame,
+    column: str,
+    by: Optional[str] = None,
+    violin: bool = False,
+    title: Optional[str] = None,
+    style: StyleTemplate = DISTRIBUTION_STYLE_TEMPLATE,
+    figsize: Tuple[float, float] = (10, 6),
+) -> Figure:
+    """Plot box or violin charts on a new figure."""
+    return _wrap_aplot(
+        aplot_box_violin,
+        pd_df=pd_df,
+        figsize=figsize,
+        column=column,
+        by=by,
+        violin=violin,
+        title=title,
+        style=style,
+    )