blob: 529203c2ab6e4a8578fb0ca3cfe9644e0c1d5e1b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
from cPickle import load
from csv import reader
from data2 import parse
import sys
def parse_row(row):
return set(e for e in map(parse, row[2:]) if e)
if __name__ == "__main__":
nodes, edges, events, event_edges = load(open("data2.pickle", "rb"))
with open(sys.argv[1]) as fh:
fh.readline()
reader = reader(fh)
d = {parse(row[1]): parse_row(row) for row in reader}
d = {k: v for (k, v) in d.iteritems() if v}
for k in d:
print len(d[k])
|
