First commit of MAMMULT code

1 files changed, 37 insertions, 0 deletions

diff --git a/structure/metrics/cartography_from_deg_vectors.py b/structure/metrics/cartography_from_deg_vectors.py
new file mode 100644
index 0000000..c40d701
--- /dev/null
+++ b/structure/metrics/cartography_from_deg_vectors.py
@@ -0,0 +1,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
+
+