diff options
| author | jeanpouget-abadie <jean.pougetabadie@gmail.com> | 2014-11-29 16:56:30 -0500 |
|---|---|---|
| committer | jeanpouget-abadie <jean.pougetabadie@gmail.com> | 2014-11-29 16:56:30 -0500 |
| commit | 3cbe733eb6ea3ee2a7b9b28319da6f2145ab4e46 (patch) | |
| tree | 327320a2f68b84a3eef5f8118d585eb1a00b1c8d /jpa_test/cascade_creation.py | |
| parent | 496d7e5b022dfc005bc2f697a62522d7022519cd (diff) | |
| download | cascades-3cbe733eb6ea3ee2a7b9b28319da6f2145ab4e46.tar.gz | |
zip -> izip
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 |