mxlpy 0.18.0__py3-none-any.whl → 0.20.0__py3-none-any.whl

mxlpy/plot.py

@@ -19,10 +19,41 @@ Functions:
 from __future__ import annotations
 
 import contextlib
+import itertools as it
+import math
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Any, Literal, cast, overload
 
+import numpy as np
+import pandas as pd
+import seaborn as sns
 from cycler import cycler
+from matplotlib import pyplot as plt
+from matplotlib.axes import Axes
+from matplotlib.colors import (
+    LogNorm,
+    Normalize,
+    SymLogNorm,
+    colorConverter,  # type: ignore
+)
+from matplotlib.figure import Figure
+from mpl_toolkits.mplot3d import Axes3D
+
+from mxlpy.label_map import LabelMapper
+
+if TYPE_CHECKING:
+    from collections.abc import Generator, Iterable, Iterator
+
+    from matplotlib.collections import QuadMesh
+    from numpy.typing import NDArray
+
+    from mxlpy.linear_label_map import LinearLabelMapper
+    from mxlpy.model import Model
+    from mxlpy.types import Array, ArrayLike
+
 
 __all__ = [
+    "Axs",
     "Color",
     "FigAx",
     "FigAxs",
@@ -31,7 +62,10 @@ __all__ = [
     "RGBA",
     "add_grid",
     "bars",
+    "bars_autogrouped",
+    "bars_grouped",
     "context",
+    "grid_labels",
     "grid_layout",
     "heatmap",
     "heatmap_from_2d_idx",
@@ -53,37 +87,45 @@ __all__ = [
     "violins_from_2d_idx",
 ]
 
-import itertools as it
-import math
-from typing import TYPE_CHECKING, Any, Literal, cast
 
-import numpy as np
-import pandas as pd
-import seaborn as sns
-from matplotlib import pyplot as plt
-from matplotlib.axes import Axes
-from matplotlib.colors import (
-    LogNorm,
-    Normalize,
-    SymLogNorm,
-    colorConverter,  # type: ignore
-)
-from matplotlib.figure import Figure
-from mpl_toolkits.mplot3d import Axes3D
+@dataclass
+class Axs:
+    """Convenience container  axes."""
 
-from mxlpy.label_map import LabelMapper
+    axs: NDArray[np.object_]
 
-if TYPE_CHECKING:
-    from collections.abc import Generator, Iterable
+    def __iter__(self) -> Iterator[Axes]:
+        """Get flat axes."""
+        yield from cast(list[Axes], self.axs.flatten())
 
-    from matplotlib.collections import QuadMesh
+    def __len__(self) -> int:
+        """Length of axes."""
+        return len(self.axs.flatten())
+
+    @overload
+    def __getitem__(self, row_col: int) -> Axes: ...
+
+    @overload
+    def __getitem__(self, row_col: slice) -> NDArray[np.object_]: ...
+
+    @overload
+    def __getitem__(self, row_col: tuple[int, int]) -> Axes: ...
+
+    @overload
+    def __getitem__(self, row_col: tuple[slice, int]) -> NDArray[np.object_]: ...
+
+    @overload
+    def __getitem__(self, row_col: tuple[int, slice]) -> NDArray[np.object_]: ...
+
+    def __getitem__(
+        self, row_col: int | slice | tuple[int | slice, int | slice]
+    ) -> Axes | NDArray[np.object_]:
+        """Get Axes or Array of Axes."""
+        return cast(Axes, self.axs[row_col])
 
-    from mxlpy.linear_label_map import LinearLabelMapper
-    from mxlpy.model import Model
-    from mxlpy.types import Array, ArrayLike
 
 type FigAx = tuple[Figure, Axes]
-type FigAxs = tuple[Figure, list[Axes]]
+type FigAxs = tuple[Figure, Axs]
 
 type Linestyle = Literal[
     "solid",
@@ -97,6 +139,7 @@ type RGB = tuple[float, float, float]
 type RGBA = tuple[float, float, float, float]
 type Color = str | RGB | RGBA
 
+
 ##########################################################################
 # Helpers
 ##########################################################################
@@ -158,7 +201,12 @@ def _partition_by_order_of_magnitude(s: pd.Series) -> list[list[str]]:
     """Partition a series into groups based on the order of magnitude of the values."""
     return [
         i.to_list()
-        for i in np.floor(np.log10(s)).to_frame(name=0).groupby(0)[0].groups.values()  # type: ignore
+        for i in s.abs()
+        .apply(np.log10)
+        .apply(np.floor)
+        .to_frame(name=0)
+        .groupby(0)[0]
+        .groups.values()  # type: ignore
     ]
 
 
@@ -258,6 +306,18 @@ def add_grid(ax: Axes) -> Axes:
     return ax
 
 
+def grid_labels(
+    axs: Axs,
+    xlabel: str | None = None,
+    ylabel: str | None = None,
+) -> None:
+    """Apply labels to left and bottom axes."""
+    for ax in axs[-1, :]:
+        ax.set_xlabel(xlabel)
+    for ax in axs[:, 0]:
+        ax.set_ylabel(ylabel)
+
+
 def rotate_xlabels(
     ax: Axes,
     rotation: float = 45,
@@ -367,7 +427,6 @@ def _default_fig_ax(
 
 
 def _default_fig_axs(
-    axs: list[Axes] | None,
     *,
     ncols: int,
     nrows: int,
@@ -391,19 +450,16 @@ def _default_fig_axs(
         Figure and Axes objects for the plot.
 
     """
-    if axs is None or len(axs) == 0:
-        fig, axs_array = plt.subplots(
-            nrows=nrows,
-            ncols=ncols,
-            sharex=sharex,
-            sharey=sharey,
-            figsize=figsize,
-            squeeze=False,
-            layout="constrained",
-        )
-        axs = list(axs_array.flatten())
-    else:
-        fig = cast(Figure, axs[0].get_figure())
+    fig, axs_array = plt.subplots(
+        nrows=nrows,
+        ncols=ncols,
+        sharex=sharex,
+        sharey=sharey,
+        figsize=figsize,
+        squeeze=False,
+        layout="constrained",
+    )
+    axs = Axs(axs_array)
 
     if grid:
         for ax in axs:
@@ -433,7 +489,6 @@ def two_axes(
 ) -> FigAxs:
     """Create a figure with two axes."""
     return _default_fig_axs(
-        None,
         ncols=2,
         nrows=1,
         figsize=figsize,
@@ -448,18 +503,17 @@ def grid_layout(
     *,
     n_cols: int = 2,
     col_width: float = 3,
-    row_height: float = 4,
+    row_height: float = 2.5,
     sharex: bool = True,
     sharey: bool = False,
     grid: bool = True,
-) -> tuple[Figure, list[Axes]]:
+) -> FigAxs:
     """Create a grid layout for the given number of groups."""
     n_cols = min(n_groups, n_cols)
     n_rows = math.ceil(n_groups / n_cols)
     figsize = (n_cols * col_width, n_rows * row_height)
 
     return _default_fig_axs(
-        None,
         ncols=n_cols,
         nrows=n_rows,
         figsize=figsize,
@@ -475,19 +529,103 @@ def grid_layout(
 
 
 def bars(
-    x: pd.DataFrame,
+    x: pd.Series | pd.DataFrame,
     *,
     ax: Axes | None = None,
     grid: bool = True,
+    xlabel: str | None = None,
+    ylabel: str | None = None,
 ) -> FigAx:
     """Plot multiple lines on the same axis."""
     fig, ax = _default_fig_ax(ax=ax, grid=grid)
-    sns.barplot(data=x, ax=ax)
-    _default_labels(ax, xlabel=x.index.name, ylabel=None)
-    ax.legend(x.columns)
+    sns.barplot(data=cast(pd.DataFrame, x), ax=ax)
+
+    if xlabel is None:
+        xlabel = x.index.name if x.index.name is not None else ""
+    _default_labels(ax, xlabel=xlabel, ylabel=ylabel)
+    if isinstance(x, pd.DataFrame):
+        ax.legend(x.columns)
     return fig, ax
 
 
+def bars_grouped(
+    groups: list[pd.DataFrame] | list[pd.Series],
+    *,
+    n_cols: int = 2,
+    col_width: float = 3,
+    row_height: float = 4,
+    sharey: bool = False,
+    grid: bool = True,
+    xlabel: str | None = None,
+    ylabel: str | None = None,
+) -> FigAxs:
+    """Plot multiple groups of lines on separate axes."""
+    fig, axs = grid_layout(
+        len(groups),
+        n_cols=n_cols,
+        col_width=col_width,
+        row_height=row_height,
+        sharex=False,
+        sharey=sharey,
+        grid=grid,
+    )
+
+    for group, ax in zip(
+        groups,
+        axs,
+        strict=False,
+    ):
+        bars(
+            group,
+            ax=ax,
+            grid=grid,
+            xlabel=xlabel,
+            ylabel=ylabel,
+        )
+
+    axsl = list(axs)
+    for i in range(len(groups), len(axs)):
+        axsl[i].set_visible(False)
+
+    return fig, axs
+
+
+def bars_autogrouped(
+    s: pd.Series | pd.DataFrame,
+    *,
+    n_cols: int = 2,
+    col_width: float = 4,
+    row_height: float = 3,
+    max_group_size: int = 6,
+    grid: bool = True,
+    xlabel: str | None = None,
+    ylabel: str | None = None,
+) -> FigAxs:
+    """Plot a series or dataframe with lines grouped by order of magnitude."""
+    group_names = _split_large_groups(
+        _partition_by_order_of_magnitude(s)
+        if isinstance(s, pd.Series)
+        else _partition_by_order_of_magnitude(s.max()),
+        max_size=max_group_size,
+    )
+
+    groups: list[pd.Series] | list[pd.DataFrame] = (
+        [s.loc[group] for group in group_names]
+        if isinstance(s, pd.Series)
+        else [s.loc[:, group] for group in group_names]
+    )
+
+    return bars_grouped(
+        groups,
+        n_cols=n_cols,
+        col_width=col_width,
+        row_height=row_height,
+        grid=grid,
+        xlabel=xlabel,
+        ylabel=ylabel,
+    )
+
+
 def lines(
     x: pd.DataFrame | pd.Series,
     *,
@@ -498,6 +636,8 @@ def lines(
     legend: bool = True,
     linewidth: float | None = None,
     linestyle: Linestyle | None = None,
+    xlabel: str | None = None,
+    ylabel: str | None = None,
 ) -> FigAx:
     """Plot multiple lines on the same axis."""
     fig, ax = _default_fig_ax(ax=ax, grid=grid)
@@ -509,7 +649,11 @@ def lines(
         linestyle=linestyle,
         color=color,
     )
-    _default_labels(ax, xlabel=x.index.name, ylabel=None)
+    _default_labels(
+        ax,
+        xlabel=x.index.name if xlabel is None else xlabel,
+        ylabel=ylabel,
+    )
     if legend:
         names = x.columns if isinstance(x, pd.DataFrame) else [str(x.name)]
         for line, name in zip(_lines, names, strict=True):
@@ -533,6 +677,8 @@ def lines_grouped(
     sharex: bool = True,
     sharey: bool = False,
     grid: bool = True,
+    xlabel: str | None = None,
+    ylabel: str | None = None,
     color: Color | list[list[Color]] | None = None,
     linewidth: float | None = None,
     linestyle: Linestyle | None = None,
@@ -561,10 +707,13 @@ def lines_grouped(
             color=color_,
             linewidth=linewidth,
             linestyle=linestyle,
+            xlabel=xlabel,
+            ylabel=ylabel,
         )
 
+    axsl = list(axs)
     for i in range(len(groups), len(axs)):
-        axs[i].set_visible(False)
+        axsl[i].set_visible(False)
 
     return fig, axs
 
@@ -577,6 +726,8 @@ def line_autogrouped(
     row_height: float = 3,
     max_group_size: int = 6,
     grid: bool = True,
+    xlabel: str | None = None,
+    ylabel: str | None = None,
     color: Color | list[list[Color]] | None = None,
     linewidth: float | None = None,
     linestyle: Linestyle | None = None,
@@ -604,6 +755,8 @@ def line_autogrouped(
         color=color,
         linestyle=linestyle,
         linewidth=linewidth,
+        xlabel=xlabel,
+        ylabel=ylabel,
     )
 
 
@@ -712,11 +865,11 @@ def _create_heatmap(
     if title is not None:
         ax.set_title(title)
     ax.set_xticks(
-        np.arange(0, len(df.columns), 1) + 0.5,
+        np.arange(0, len(df.columns), 1, dtype=float) + 0.5,
         labels=xticklabels,
     )
     ax.set_yticks(
-        np.arange(0, len(df.index), 1) + 0.5,
+        np.arange(0, len(df.index), 1, dtype=float) + 0.5,
         labels=yticklabels,
     )
 
@@ -886,7 +1039,7 @@ def violins_from_2d_idx(
         grid=grid,
     )
 
-    for ax, col in zip(axs[: len(df.columns)], df.columns, strict=True):
+    for ax, col in zip(axs[: len(df.columns)].flatten(), df.columns, strict=True):
         ax.set_title(col)
         violins(df[col].unstack(), ax=ax)
 

mxlpy/report.py

@@ -1,5 +1,7 @@
 """Generate a report comparing two models."""
 
+from __future__ import annotations
+
 from collections.abc import Callable
 from datetime import UTC, datetime
 from pathlib import Path
@@ -10,7 +12,10 @@ import sympy
 from mxlpy.meta.source_tools import fn_to_sympy
 from mxlpy.model import Model
 
-__all__ = ["AnalysisFn", "markdown"]
+__all__ = [
+    "AnalysisFn",
+    "markdown",
+]
 
 type AnalysisFn = Callable[[Model, Model, Path], tuple[str, Path]]
 
@@ -84,9 +89,29 @@ def markdown(
 
     """
     content: list[str] = [
-        f"# Report:  {datetime.now(UTC).strftime('%Y-%m-%d')}",
+        f"# Report:  {datetime.now(UTC).strftime('%Y-%m-%d')}\n",
     ]
 
+    # Unused
+    if unused := m2.get_unused_parameters():
+        content.append("## <span style='color: red'>Unused parameters</span>\n")
+        names = "\n".join(f"<li>{i}</li>\n" for i in sorted(unused))
+        content.append(f"<ul>\n{names}\n</ul>\n")
+
+    # Model stats
+    content.extend(
+        [
+            "| Model component | Old | New |",
+            "| --- | --- | --- |",
+            f"| variables | {len(m1.variables)} | {len(m2.variables)}|",
+            f"| parameters | {len(m1.parameters)} | {len(m2.parameters)}|",
+            f"| derived parameters | {len(m1.derived_parameters)} | {len(m2.derived_parameters)}|",
+            f"| derived variables | {len(m1.derived_variables)} | {len(m2.derived_variables)}|",
+            f"| reactions | {len(m1.reactions)} | {len(m2.reactions)}|",
+            f"| surrogates | {len(m1._surrogates)} | {len(m2._surrogates)}|",  # noqa: SLF001
+        ]
+    )
+
     # Variables
     new_variables, removed_variables, changed_variables = _new_removed_changed(
         m1.variables, m2.variables
@@ -106,7 +131,7 @@ def markdown(
     if len(variables) >= 1:
         content.extend(
             (
-                "## Variables\n",
+                "## Variables\n\n",
                 "| Name | Old Value | New Value |",
                 "| ---- | --------- | --------- |",
             )
@@ -132,7 +157,7 @@ def markdown(
     if len(pars) >= 1:
         content.extend(
             (
-                "## Parameters\n",
+                "## Parameters\n\n",
                 "| Name | Old Value | New Value |",
                 "| ---- | --------- | --------- |",
             )
@@ -159,7 +184,7 @@ def markdown(
     if len(derived) >= 1:
         content.extend(
             (
-                "## Derived\n",
+                "## Derived\n\n",
                 "| Name | Old Value | New Value |",
                 "| ---- | --------- | --------- |",
             )
@@ -187,7 +212,7 @@ def markdown(
     if len(reactions) >= 1:
         content.extend(
             (
-                "## Reactions\n",
+                "## Reactions\n\n",
                 "| Name | Old Value | New Value |",
                 "| ---- | --------- | --------- |",
             )
@@ -207,7 +232,7 @@ def markdown(
     if len(dependent) >= 1:
         content.extend(
             (
-                "## Numerical differences of dependent values\n",
+                "## Numerical differences of dependent values\n\n",
                 "| Name | Old Value | New Value | Relative Change | ",
                 "| ---- | --------- | --------- | --------------- | ",
             )
@@ -226,7 +251,7 @@ def markdown(
     if len(rhs) >= 1:
         content.extend(
             (
-                "## Numerical differences of right hand side values\n",
+                "## Numerical differences of right hand side values\n\n",
                 "| Name | Old Value | New Value | Relative Change | ",
                 "| ---- | --------- | --------- | --------------- | ",
             )

mxlpy/sbml/__init__.py

@@ -5,10 +5,10 @@ Allows importing and exporting metabolic models in SBML format.
 
 from __future__ import annotations
 
+from ._export import write
+from ._import import read
+
 __all__ = [
     "read",
     "write",
 ]
-
-from ._export import write
-from ._import import read

mxlpy/sbml/_data.py

@@ -1,5 +1,12 @@
 from __future__ import annotations
 
+from dataclasses import dataclass
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from collections.abc import Mapping
+
+
 __all__ = [
     "AtomicUnit",
     "Compartment",
@@ -11,12 +18,6 @@ __all__ = [
     "Reaction",
 ]
 
-from dataclasses import dataclass
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from collections.abc import Mapping
-
 
 @dataclass
 class AtomicUnit:

mxlpy/sbml/_export.py

@@ -447,7 +447,14 @@ def _create_sbml_variables(
         cpd.setConstant(False)
         cpd.setBoundaryCondition(False)
         cpd.setHasOnlySubstanceUnits(False)
-        cpd.setInitialAmount(float(value))
+        if isinstance(value, Derived):
+            ar = sbml_model.createInitialAssignment()
+            ar.setId(_convert_id_to_sbml(id_=name, prefix="IA"))
+            ar.setName(_convert_id_to_sbml(id_=name, prefix="IA"))
+            ar.setVariable(_convert_id_to_sbml(id_=name, prefix="IA"))
+            ar.setMath(_sbmlify_fn(value.fn, value.args))
+        else:
+            cpd.setInitialAmount(float(value))
 
 
 def _create_sbml_derived_variables(*, model: Model, sbml_model: libsbml.Model) -> None:

mxlpy/sbml/_mathml.py

@@ -1,5 +1,13 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING, Any
+
+from ._name_conversion import _name_to_py
+from ._unit_conversion import get_ast_types
+
+if TYPE_CHECKING:
+    from libsbml import ASTNode
+
 __all__ = [
     "AST_TYPES",
     "handle_ast_constant_e",
@@ -73,13 +81,6 @@ __all__ = [
     "parse_sbml_math",
 ]
 
-from typing import TYPE_CHECKING, Any
-
-from ._name_conversion import _name_to_py
-from ._unit_conversion import get_ast_types
-
-if TYPE_CHECKING:
-    from libsbml import ASTNode
 
 AST_TYPES = get_ast_types()
 

mxlpy/sbml/_name_conversion.py

@@ -3,7 +3,11 @@ from __future__ import annotations
 import keyword
 import re
 
-__all__ = ["RE_FROM_SBML", "RE_KWDS", "SBML_DOT"]
+__all__ = [
+    "RE_FROM_SBML",
+    "RE_KWDS",
+    "SBML_DOT",
+]
 
 RE_KWDS = re.compile("|".join(f"^{i}$" for i in keyword.kwlist))
 SBML_DOT = "__SBML_DOT__"

mxlpy/scan.py

@@ -15,19 +15,6 @@ Functions:
 
 from __future__ import annotations
 
-from mxlpy.integrators import DefaultIntegrator
-
-__all__ = [
-    "ProtocolWorker",
-    "SteadyStateWorker",
-    "TimeCourse",
-    "TimeCourseWorker",
-    "TimePoint",
-    "steady_state",
-    "time_course",
-    "time_course_over_protocol",
-]
-
 from dataclasses import dataclass
 from functools import partial
 from typing import TYPE_CHECKING, Protocol, Self, cast
@@ -35,14 +22,10 @@ from typing import TYPE_CHECKING, Protocol, Self, cast
 import numpy as np
 import pandas as pd
 
+from mxlpy.integrators import DefaultIntegrator
 from mxlpy.parallel import Cache, parallelise
 from mxlpy.simulator import Result, Simulator
-from mxlpy.types import (
-    IntegratorType,
-    ProtocolByPars,
-    SteadyStates,
-    TimeCourseByPars,
-)
+from mxlpy.types import IntegratorType, ProtocolByPars, SteadyStates, TimeCourseByPars
 
 if TYPE_CHECKING:
     from collections.abc import Callable
@@ -51,6 +34,18 @@ if TYPE_CHECKING:
     from mxlpy.types import Array
 
 
+__all__ = [
+    "ProtocolWorker",
+    "SteadyStateWorker",
+    "TimeCourse",
+    "TimeCourseWorker",
+    "TimePoint",
+    "steady_state",
+    "time_course",
+    "time_course_over_protocol",
+]
+
+
 def _update_parameters_and_initial_conditions[T](
     pars: pd.Series,
     fn: Callable[[Model], T],