summaryrefslogtreecommitdiff
path: root/structure/metrics/cartography_from_deg_vectors.py
blob: c40d701403c448acf9b32fd8fb2c8cd817cff52a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
####
##
## Take as input a file containing, on each line, the degree vector of
## a node of the multiplex, and compute the multiplex cartography
## diagram
##
##

import sys

if len(sys.argv) < 2:
    print "Usage: %s <node_deg_vectors>" % sys.argv[0]
    sys.exit(1)

filein=sys.argv[1]

M = -1

with open(filein,"r") as lines:
    for l in lines:
        if l[0] == "#":
            continue
        elems = [int(x) for x in l.strip(" \n").split(" ")]
        if (M == -1):
            M = len(elems)
        sum_elems = 0
        part = 0
        for val in elems:
            sum_elems += val
            part += val*val
        if sum_elems > 0:
            part = M * 1.0 / (M -1) * (1 - part * 1.0 / (sum_elems * sum_elems))
        else:
            part = 0.0
        print sum_elems, part