phylogenie 2.1.14__py3-none-any.whl → 2.1.16__py3-none-any.whl

phylogenie/plot.py

@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
 from matplotlib.axes import Axes
 
 from phylogenie.tree import Tree
-from phylogenie.utils import get_node_depths
+from phylogenie.utils import get_node_depth_levels, get_node_depths
 
 
 class Coloring(str, Enum):
@@ -22,15 +22,14 @@ def plot_tree(
     coloring: str | Coloring | None = None,
     cmap: str | None = None,
     show_legend: bool = True,
-    x_feature: str | None = None,
 ) -> Axes:
     if ax is None:
         ax = plt.gca()
 
     xs = (
-        get_node_depths(tree)
-        if x_feature is None
-        else {node: node.get(x_feature) for node in tree}
+        get_node_depth_levels(tree)
+        if any(node.branch_length is None for node in tree)
+        else get_node_depths(tree)
     )
     ys = {node: i for i, node in enumerate(tree.inorder_traversal())}
 

phylogenie/tree.py

@@ -1,4 +1,5 @@
-from collections.abc import Iterator
+from collections import deque
+from collections.abc import Callable, Iterator
 from typing import Any
 
 
@@ -55,6 +56,10 @@ class Tree:
             return 0
         return 1 + max(child.height_level for child in self.children)
 
+    @property
+    def n_leaves(self) -> int:
+        return len(self.get_leaves())
+
     def set(self, key: str, value: Any) -> None:
         self._features[key] = value
 
@@ -102,6 +107,13 @@ class Tree:
             yield from child.postorder_traversal()
         yield self
 
+    def breadth_first_traversal(self) -> Iterator["Tree"]:
+        queue: deque["Tree"] = deque([self])
+        while queue:
+            node = queue.popleft()
+            yield node
+            queue.extend(node.children)
+
     def get_node(self, name: str) -> "Tree":
         for node in self:
             if node.name == name:
@@ -114,10 +126,10 @@ class Tree:
     def get_leaves(self) -> tuple["Tree", ...]:
         return tuple(node for node in self if node.is_leaf())
 
-    def ladderize(self, feature: str) -> None:
-        self._children.sort(key=lambda x: x.get(feature))
+    def ladderize(self, criterion: Callable[["Tree"], Any]) -> None:
+        self._children.sort(key=criterion)
         for child in self.children:
-            child.ladderize(feature)
+            child.ladderize(criterion)
 
     def copy(self):
         new_tree = Tree(self.name, self.branch_length)

{phylogenie-2.1.14.dist-info → phylogenie-2.1.16.dist-info}/METADATA

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

{phylogenie-2.1.14.dist-info → phylogenie-2.1.16.dist-info}/RECORD

@@ -10,13 +10,13 @@ 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=5InWceLwjZV2qVYv2jaTJPs_0gfAEaKYGQt_jnLpaz0,3615
+phylogenie/plot.py,sha256=I1QH7bZ09t7KtI4LleQlph9YZZeyCwu_Jr-YH2JKrW0,3618
 phylogenie/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phylogenie/skyline/__init__.py,sha256=7pF4CUb4ZCLzNYJNhOjpuTOLTRhlK7L6ugfccNqjIGo,620
 phylogenie/skyline/matrix.py,sha256=Gl8OgKjtieG0NwPYiPimKI36gefV8fm_OeorjdXxPTs,9146
 phylogenie/skyline/parameter.py,sha256=EM9qlPt0JhMBy3TbztM0dj24BaGNEy8KWKdTObDKhbI,4644
 phylogenie/skyline/vector.py,sha256=bJP7_FNX_Klt6wXqsyfj0KX3VNj6-dIhzCKSJuQcOV0,7115
-phylogenie/tree.py,sha256=PjyfsjYWadvIQSOxyIO0oaRVN2pecXEnUR-k1mM4Vic,4159
+phylogenie/tree.py,sha256=MTMm_MyJt-ExJwJXu-WUCZPZ3U_YvF4XLNqRb50cgPI,4516
 phylogenie/treesimulator/__init__.py,sha256=yqS2vtYMhdWSXc9RAnX1dd4zAqSQweMLyVKTnJLfGTU,1106
 phylogenie/treesimulator/events/__init__.py,sha256=6zSgZ0MEUMvTK4yPlSolJnRWzCARLS-jYreTzh45mQo,1033
 phylogenie/treesimulator/events/contact_tracing.py,sha256=_nJ85yhgGkeruQgMHvGpDYoyhheBf8M4LgZWiWdi5dY,4801
@@ -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=l7kJ0mkdhplz_hTUseBxaGpEKngsWUOSiCMyLMIWYZQ,1936
-phylogenie-2.1.14.dist-info/LICENSE.txt,sha256=NUrDqElK-eD3I0WqC004CJsy6cs0JgsAoebDv_42-pw,1071
-phylogenie-2.1.14.dist-info/METADATA,sha256=elXtIXUWfGalXgdhTgretrdYVQwOWwiSfHUiZicf5WQ,5477
-phylogenie-2.1.14.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-phylogenie-2.1.14.dist-info/entry_points.txt,sha256=Rt6_usN0FkBX1ZfiqCirjMN9FKOgFLG8rydcQ8kugeE,51
-phylogenie-2.1.14.dist-info/RECORD,,
+phylogenie-2.1.16.dist-info/LICENSE.txt,sha256=NUrDqElK-eD3I0WqC004CJsy6cs0JgsAoebDv_42-pw,1071
+phylogenie-2.1.16.dist-info/METADATA,sha256=2Fyg0A7W9ooDxloKDdi6ZYg31fGMpRKYnDHwjgMjCGI,5477
+phylogenie-2.1.16.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+phylogenie-2.1.16.dist-info/entry_points.txt,sha256=Rt6_usN0FkBX1ZfiqCirjMN9FKOgFLG8rydcQ8kugeE,51
+phylogenie-2.1.16.dist-info/RECORD,,