Home

A Python library for partitioning and aggregation of spatial network graphs, with a special focus in electrical power systems

Get Started

GitHub Repository

Quick Start ✨

import npap
from npap import AggregationMode

# 1. Initialize the main manager
manager = npap.PartitionAggregatorManager()

# 2. Load data
manager.load_data(strategy="networkx_direct", graph=your_graph)

# 2.1. Aggregate parallel edges (if applicable)
manager.aggregate_parallel_edges(
    edge_properties={"p_max": "average", "x": "equivalent_reactance"},
    default_strategy="average",
    warn_on_defaults=False,
)

# 3. Create partition result
partition_result = manager.partition(strategy="geographical_kmeans", n_clusters=250)

# 3.1. Plot the partitioned network
manager.plot_network(style="clustered", title="Partitioned Network")

# 4. Aggregate the network based on a given AggregationMode
aggregated_network = manager.aggregate(mode=AggregationMode.GEOGRAPHICAL)