Added Step counter to the gif.

Fixed multiple infected nodes at step 0.
2022-06-20 01:10:32 +02:00 · 2022-06-20 01:10:32 +02:00 · 02800bcb7e
commit 02800bcb7e
parent eb5e546e73
1 changed files with 4 additions and 1 deletions
--- a/network_attack_propagation.py
+++ b/network_attack_propagation.py
@ -94,9 +94,12 @@ def update(num, layout, g_repr, ax, our_graph: Graph):
    if not our_graph.is_alive():
        return

+    if num != 0:
        our_graph.infect_step()
    ax.clear()

+    ax.set_title(f'Step: {num}', loc='right', fontsize=30)
+
    colors = ['red' if n.is_infected else 'blue' for n in g_repr]
    sizes = [200 if n.is_infected else 50 for n in g_repr]
    nx.draw(