diff options
Diffstat (limited to 'jpa_test/cascade_creation.py')
| -rw-r--r-- | jpa_test/cascade_creation.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/jpa_test/cascade_creation.py b/jpa_test/cascade_creation.py index 32d9c4a..80054a7 100644 --- a/jpa_test/cascade_creation.py +++ b/jpa_test/cascade_creation.py @@ -2,6 +2,8 @@ import networkx as nx import numpy as np import collections +from itertools import izip + class InfluenceGraph(nx.Graph): """ networkX graph with mat and logmat attributes @@ -50,7 +52,7 @@ class Cascade(list): Returns lists of infections times for node i in cascade """ infected_times = [] - for t, infected_set in zip(xrange(len(self)), self): + for t, infected_set in izip(xrange(len(self)), self): if infected_set[node]: infected_times.append(t) return infected_times |