diff options
| author | Thibaut Horel <thibaut.horel@gmail.com> | 2015-09-14 23:08:02 -0400 |
|---|---|---|
| committer | Thibaut Horel <thibaut.horel@gmail.com> | 2015-09-14 23:08:02 -0400 |
| commit | ab0b1f3cefedb35327a19ec1b6afd560bfdf802d (patch) | |
| tree | b777f3e2c0ac0e712d8c5faab5107b1d236e2c3a /experiments/build_network.py | |
| parent | 960676226862d2d68c7a9c04c56d4f8157803025 (diff) | |
| download | criminal_cascades-ab0b1f3cefedb35327a19ec1b6afd560bfdf802d.tar.gz | |
Import supplements and repo reorganization
Diffstat (limited to 'experiments/build_network.py')
| -rw-r--r-- | experiments/build_network.py | 58 |
1 files changed, 0 insertions, 58 deletions
diff --git a/experiments/build_network.py b/experiments/build_network.py deleted file mode 100644 index c58d5cf..0000000 --- a/experiments/build_network.py +++ /dev/null @@ -1,58 +0,0 @@ -from csv import DictReader -import sys -from cPickle import dump -from os.path import splitext - - -def build_network(filename): - victims = {} - non_victims = {} - age = 0 - t_max = 1000 - with open(filename) as fh: - reader = DictReader(fh) - for row in reader: - from_, to = int(float(row["from"])), int(float(row["to"])) - dist = int(row["dist"]) - w1, w2, w3 = float(row["w1"]), float(row["w2"]), float(row["w3"]) - # if int(float(row["dist"])) > 1: - # continue - # 'to' is a victim - if row["t2"] != "NA": - dt = int(row["t2"]) - int(row["t1"]) - parent = (from_, dist, dt, w1, w2, w3) - if to not in victims: - age += int(row["t2"]) - int(row["spawn2"]) - victims[to] = [] - victims[to].append(parent) - if from_ not in victims: - age += int(row["t1"]) - int(row["spawn1"]) - victims[from_] = [] - # 'to' is not a victim - else: - dt = t_max - int(row["t1"]) - parent = (from_, dist, dt, w1, w2, w3) - if to not in non_victims: - # age += 3012 - int(row["spawn2"]) - non_victims[to] = [] - non_victims[to].append(parent) - if from_ not in victims: - age += int(row["t1"]) - int(row["spawn1"]) - victims[from_] = [] - root_victims = {} - for victim in victims.keys(): - if not victims[victim]: - del victims[victim] - root_victims[victim] = [] - print len(root_victims), len(victims), len(non_victims) - return root_victims, victims, non_victims, age - - -if __name__ == "__main__": - if len(sys.argv) < 2: - sys.exit("usage: {0} <file>".format(sys.argv[0])) - - filename = sys.argv[1] - root, _ = splitext(filename) - root_victims, victims, non_victims, age = build_network(filename) - dump((root_victims, victims, non_victims, age), open(root + ".pickle", "w")) |
