Lollipop graph

Example I - Simple Lollipop using networkx

import networkx as nx

import pop2net as p2n

model = p2n.Model()
creator = p2n.Creator(model)
inspector = p2n.NetworkInspector(model)


class Lollipop(p2n.LocationDesigner):
    n_agents = 7
    nxgraph = nx.lollipop_graph(5, 3)


for _ in range(20):
    p2n.Agent(model)

creator.create_locations(
    location_designers=[
        Lollipop,
    ]
)

LocationList (36 objects)

inspector.plot_bipartite_network()

inspector.plot_agent_network()

Example II - A Lollipop flower

model = p2n.Model()
creator = p2n.Creator(model)
inspector = p2n.NetworkInspector(model)


class Lollipop(p2n.LocationDesigner):
    n_locations = 5
    nxgraph = nx.lollipop_graph(5, 3)
    only_exact_n_agents = True


class Center(p2n.LocationDesigner):
    def filter(self, agent):
        return agent.Lollipop_tail or not agent.Lollipop

    def bridge(self, agent):
        return agent.Lollipop_assigned


for _ in range(41):
    p2n.Agent(model)

creator.create_locations(
    location_designers=[
        Lollipop,
        Center,
    ]
)

LocationList (70 objects)

inspector.plot_bipartite_network()

inspector.plot_agent_network()