s444409/Network_attack_propagation
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Added Step counter to the gif.

Browse Source 
Fixed multiple infected nodes at step 0.
This commit is contained in:
Krystian Wasilewski 2022-06-20 01:10:32 +02:00
parent eb5e546e73
commit 02800bcb7e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
network_attack_propagation.py
View File

@ -94,9 +94,12 @@ def update(num, layout, g_repr, ax, our_graph: Graph):
if not our_graph.is_alive(): if not our_graph.is_alive():
return return
our_graph.infect_step() if num != 0:
our_graph.infect_step()
ax.clear() 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] 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] sizes = [200 if n.is_infected else 50 for n in g_repr]
nx.draw( nx.draw(