MatplotLibAPI 4.0.3__tar.gz → 4.0.4__tar.gz

{matplotlibapi-4.0.3 → matplotlibapi-4.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: MatplotLibAPI
-Version: 4.0.3
+Version: 4.0.4
 License-File: LICENSE
 Requires-Python: >=3.9
 Requires-Dist: kaleido

{matplotlibapi-4.0.3 → matplotlibapi-4.0.4}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 [project]
 name = "MatplotLibAPI"
-version = "4.0.3"
+version = "4.0.4"
 readme = "README.md"
 requires-python = ">=3.9"
 dependencies = [

{matplotlibapi-4.0.3 → matplotlibapi-4.0.4}/src/MatplotLibAPI/accessor.py

@@ -1278,6 +1278,57 @@ class DataFrameAccessor:
             **kwargs,
         )
 
+    def fplot_network(
+        self,
+        edge_source_col: str = "source",
+        edge_target_col: str = "target",
+        edge_weight_col: str = "weight",
+        title: Optional[str] = None,
+        style: Optional[StyleTemplate] = None,
+        layout_seed: Optional[int] = None,
+        figsize: Tuple[float, float] = FIG_SIZE,
+    ) -> Figure:
+        """Plot a network graph on a new figure.
+
+        Parameters
+        ----------
+        edge_source_col : str, optional
+            Column for source nodes. The default is "source".
+        edge_target_col : str, optional
+            Column for target nodes. The default is "target".
+        edge_weight_col : str, optional
+            Column for edge weights. The default is "weight".
+        title : str, optional
+            Chart title.
+        style : StyleTemplate, optional
+            Styling template. The default is `NETWORK_STYLE_TEMPLATE`.
+        layout_seed : int, optional
+            Seed forwarded to the spring layout. The default is ``_DEFAULT["SPRING_LAYOUT_SEED"]``.
+        figsize : tuple[float, float], optional
+            Figure size. The default is FIG_SIZE.
+
+        Returns
+        -------
+        Figure
+            The new Matplotlib figure with the plot.
+        """
+        kwargs: Dict[str, Any] = {}
+        if layout_seed is not None:
+            kwargs["layout_seed"] = layout_seed
+
+        from .network import NETWORK_STYLE_TEMPLATE, fplot_network
+
+        return fplot_network(
+            pd_df=self._obj,
+            edge_source_col=edge_source_col,
+            edge_target_col=edge_target_col,
+            edge_weight_col=edge_weight_col,
+            title=title,
+            style=style or NETWORK_STYLE_TEMPLATE,
+            figsize=figsize,
+            **kwargs,
+        )
+
     def aplot_network_components(
         self,
         edge_source_col: str = "source",

{matplotlibapi-4.0.3 → matplotlibapi-4.0.4}/src/MatplotLibAPI/network/core.py

@@ -453,6 +453,33 @@ class NetworkGraph(BasePlot):
         """Return the degree assortativity coefficient of the graph."""
         return nx.degree_assortativity_coefficient(self._nx_graph)
 
+    @property
+    def degree_distribution(self) -> Dict[int, int]:
+        """Return the count of nodes for each degree.
+
+        Returns
+        -------
+        dict[int, int]
+            Mapping from node degree to number of nodes with that degree.
+        """
+        distribution: Dict[int, int] = defaultdict(int)
+        for _, degree in self._nx_graph.degree():
+            distribution[int(degree)] += 1
+        return dict(sorted(distribution.items()))
+
+    @property
+    def degree_sequence(self) -> List[int]:
+        """Return node degrees sorted in descending order.
+
+        Returns
+        -------
+        list[int]
+            Degree for each node sorted from highest to lowest.
+        """
+        return sorted(
+            (int(degree) for _, degree in self._nx_graph.degree()), reverse=True
+        )
+
     def add_node(self, node: Any, **attributes: Any):
         """Add a node with optional attributes.
 

{matplotlibapi-4.0.3 → matplotlibapi-4.0.4}/src/MatplotLibAPI/network/plot.py

@@ -501,6 +501,8 @@ def fplot_network(
     ).fplot(
         title=title,
         style=style,
+        edge_weight_col=edge_weight_col,
+        layout_seed=layout_seed,
         figsize=figsize,
     )
 

{matplotlibapi-4.0.3 → matplotlibapi-4.0.4}/tests/test_network.py

@@ -83,6 +83,44 @@ def test_accessor_fplot_network_components(load_sample_df):
     assert isinstance(fig, Figure)
 
 
+def test_accessor_fplot_network(load_sample_df):
+    """Render a network figure via the pandas accessor."""
+
+    df = load_sample_df("network.csv")
+
+    fig = df.mpl.fplot_network(
+        edge_source_col="city_a",
+        edge_target_col="city_b",
+        edge_weight_col="distance_km",
+    )
+
+    assert isinstance(fig, Figure)
+
+
+def test_accessor_fplot_network_forwards_layout_seed(monkeypatch):
+    """Forward ``layout_seed`` from accessor calls to network plotting."""
+
+    df = pd.DataFrame(
+        {
+            "source": ["a", "b", "c"],
+            "target": ["b", "c", "a"],
+            "weight": [1, 2, 3],
+        }
+    )
+    captured_layout_seed = []
+
+    def fake_aplot(self, *args, **kwargs):  # type: ignore[override]
+        captured_layout_seed.append(kwargs.get("layout_seed"))
+        return plt.gca()
+
+    monkeypatch.setattr(NetworkGraph, "aplot", fake_aplot)
+
+    fig = df.mpl.fplot_network(layout_seed=123)
+
+    assert isinstance(fig, Figure)
+    assert captured_layout_seed == [123]
+
+
 def test_softmax_matches_expected_probabilities():
     """Return softmax probabilities consistent with NumPy operations."""
 
@@ -152,6 +190,28 @@ def test_scale_weights_respects_precomputed_deciles():
     assert _scale_weights(weights, deciles=deciles) == expected
 
 
+def test_degree_distribution_returns_node_counts_by_degree():
+    """Return the count of nodes grouped by their degree."""
+
+    graph = NetworkGraph()
+    graph.add_edge("a", "b")
+    graph.add_edge("b", "c")
+    graph.add_node("isolated")
+
+    assert graph.degree_distribution == {0: 1, 1: 2, 2: 1}
+
+
+def test_degree_sequence_returns_descending_node_degrees():
+    """Return node degrees sorted from highest to lowest."""
+
+    graph = NetworkGraph()
+    graph.add_edge("a", "b")
+    graph.add_edge("b", "c")
+    graph.add_node("isolated")
+
+    assert graph.degree_sequence == [2, 1, 1, 0]
+
+
 def test_network_layout_respects_precomputed_deciles() -> None:
     """Reuse provided deciles to produce stable layout scaling."""