phylogenie 2.1.17__py3-none-any.whl → 2.1.18__py3-none-any.whl

phylogenie/plot.py

@@ -1,9 +1,11 @@
 from enum import Enum
+from typing import Any
 
 import matplotlib.colors as mcolors
 import matplotlib.patches as mpatches
 import matplotlib.pyplot as plt
 from matplotlib.axes import Axes
+from mpl_toolkits.axes_grid1.inset_locator import inset_axes  # pyright: ignore
 
 from phylogenie.tree import Tree
 from phylogenie.utils import get_node_depth_levels, get_node_depths
@@ -22,7 +24,10 @@ def plot_tree(
     coloring: str | Coloring | None = None,
     cmap: str | None = None,
     show_legend: bool = True,
-) -> Axes:
+    show_hist: bool = True,
+    hist_kwargs: dict[str, Any] | None = None,
+    hist_axes_kwargs: dict[str, Any] | None = None,
+) -> Axes | tuple[Axes, Axes]:
     if ax is None:
         ax = plt.gca()
 
@@ -34,7 +39,8 @@ def plot_tree(
     ys = {node: i for i, node in enumerate(tree.inorder_traversal())}
 
     if color_by is not None:
-        features = set(node.get(color_by) for node in tree if color_by in node.features)
+        features = [node.get(color_by) for node in tree if color_by in node.features]
+
         if coloring is None and any(isinstance(f, float) for f in features):
             coloring = Coloring.CONTINUOUS
         elif coloring is None:
@@ -47,7 +53,7 @@ def plot_tree(
                 )
             colormap = plt.get_cmap("tab20" if cmap is None else cmap)
             feature_colors = {
-                f: mcolors.to_hex(colormap(i)) for i, f in enumerate(features)
+                f: mcolors.to_hex(colormap(i)) for i, f in enumerate(set(features))
             }
             colors = {
                 node: (
@@ -61,7 +67,7 @@ def plot_tree(
             if show_legend:
                 legend_handles = [
                     mpatches.Patch(color=feature_colors[f], label=str(f))
-                    for f in features
+                    for f in feature_colors
                 ]
                 if any(color_by not in node.features for node in tree):
                     legend_handles.append(
@@ -81,8 +87,34 @@ def plot_tree(
                 for node in tree
             }
 
-            sm = plt.cm.ScalarMappable(cmap=colormap, norm=norm)
-            ax.get_figure().colorbar(sm, ax=ax)  # pyright: ignore
+            if show_hist:
+                default_hist_axes_kwargs = {
+                    "width": "25%",
+                    "height": "25%",
+                    "loc": "lower left",
+                    "borderpad": 3,
+                }
+                if hist_axes_kwargs is not None:
+                    default_hist_axes_kwargs.update(hist_axes_kwargs)
+                hist_ax = inset_axes(ax, **default_hist_axes_kwargs)  # pyright: ignore
+
+                hist_kwargs = {} if hist_kwargs is None else hist_kwargs
+                _, bins, patches = hist_ax.hist(  # pyright: ignore
+                    features, **hist_kwargs
+                )
+
+                for patch, b0, b1 in zip(  # pyright: ignore
+                    patches, bins[:-1], bins[1:]  # pyright: ignore
+                ):
+                    midpoint = (b0 + b1) / 2  # pyright: ignore
+                    patch.set_facecolor(colormap(norm(midpoint)))  # pyright: ignore
+
+                hist_ax.set_xlabel(color_by.capitalize(), fontsize=8)  # pyright: ignore
+                hist_ax.set_ylabel("Count", fontsize=8)  # pyright: ignore
+                hist_ax.tick_params(axis="both", labelsize=6)  # pyright: ignore
+            else:
+                sm = plt.cm.ScalarMappable(cmap=colormap, norm=norm)
+                ax.get_figure().colorbar(sm, ax=ax)  # pyright: ignore
 
         else:
             raise ValueError(
@@ -100,5 +132,14 @@ def plot_tree(
         ax.vlines(x=x0, ymin=y0, ymax=y1, color=colors[node])  # pyright: ignore
         ax.hlines(y=y1, xmin=x0, xmax=x1, color=colors[node])  # pyright: ignore
 
+    for node in tree:
+        x1, y1 = xs[node], ys[node]
+        if node.parent is None:
+            ax.hlines(y=y1, xmin=0, xmax=x1, color=colors[node])  # pyright: ignore
+            continue
+        x0, y0 = xs[node.parent], ys[node.parent]
+        ax.vlines(x=x0, ymin=y0, ymax=y1, color=colors[node])  # pyright: ignore
+        ax.hlines(y=y1, xmin=x0, xmax=x1, color=colors[node])  # pyright: ignore
+
     ax.set_yticks([])  # pyright: ignore
     return ax

{phylogenie-2.1.17.dist-info → phylogenie-2.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: phylogenie
-Version: 2.1.17
+Version: 2.1.18
 Summary: Generate phylogenetic datasets with minimal setup effort
 Author: Gabriele Marino
 Author-email: gabmarino.8601@gmail.com

{phylogenie-2.1.17.dist-info → phylogenie-2.1.18.dist-info}/RECORD

@@ -10,7 +10,7 @@ phylogenie/io.py,sha256=nwy8DOknt0HqF9qMeFZHrCmSXpM5AGrU5oajwTtD6vY,3973
 phylogenie/main.py,sha256=vtvSpQxBNlYABoFQ25czl-l3fIr4QRo3svWVd-jcArw,1170
 phylogenie/models.py,sha256=pCg9ob0RpLUHwM49x4knKxL4FNPr3-EU_6zMXsvxtAg,370
 phylogenie/msa.py,sha256=JDGyZUsAq6-m-SQjoCDjAkAZIxfgyl_PDIhdYn5HOow,2064
-phylogenie/plot.py,sha256=I1QH7bZ09t7KtI4LleQlph9YZZeyCwu_Jr-YH2JKrW0,3618
+phylogenie/plot.py,sha256=MgRIquHtI2QpHR7NH7dsmfaeDN1ICU6CpZjShLqHQC0,5505
 phylogenie/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phylogenie/skyline/__init__.py,sha256=7pF4CUb4ZCLzNYJNhOjpuTOLTRhlK7L6ugfccNqjIGo,620
 phylogenie/skyline/matrix.py,sha256=Gl8OgKjtieG0NwPYiPimKI36gefV8fm_OeorjdXxPTs,9146
@@ -28,8 +28,8 @@ phylogenie/treesimulator/model.py,sha256=Nyg6R8XmMwZMSw1-dII81sU9uU7tDe-NMs8v1qK
 phylogenie/typeguards.py,sha256=JtqmbEWJZBRHbWgCvcl6nrWm3VcBfzRbklbTBYHItn0,1325
 phylogenie/typings.py,sha256=GknvAFXyiaWeeYJ8Lk5d6E2VHT-xW6ONEojYbtJYiB8,476
 phylogenie/utils.py,sha256=faDicpAtjxM-QDwPKOv8AOxPe7skPUOGKmVWBZRzUbU,1990
-phylogenie-2.1.17.dist-info/LICENSE.txt,sha256=NUrDqElK-eD3I0WqC004CJsy6cs0JgsAoebDv_42-pw,1071
-phylogenie-2.1.17.dist-info/METADATA,sha256=TUa2AtURiUDVcWfYzt_6BQYaH7N4Izgco0O5vKVOuH0,5477
-phylogenie-2.1.17.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-phylogenie-2.1.17.dist-info/entry_points.txt,sha256=Rt6_usN0FkBX1ZfiqCirjMN9FKOgFLG8rydcQ8kugeE,51
-phylogenie-2.1.17.dist-info/RECORD,,
+phylogenie-2.1.18.dist-info/LICENSE.txt,sha256=NUrDqElK-eD3I0WqC004CJsy6cs0JgsAoebDv_42-pw,1071
+phylogenie-2.1.18.dist-info/METADATA,sha256=N3F8qjtVEtqb0kxfywFd7GpTiGyCSNWwCwlUmyuNUfQ,5477
+phylogenie-2.1.18.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+phylogenie-2.1.18.dist-info/entry_points.txt,sha256=Rt6_usN0FkBX1ZfiqCirjMN9FKOgFLG8rydcQ8kugeE,51
+phylogenie-2.1.18.dist-info/RECORD,,