Import files

Overlap/edgelist_to_Graphviz.py

+#!/usr/bin/env python3
+
+import csv
+import pygraphviz as pgv
+import sys
+
+if len(sys.argv) != 3:
+    sys.exit("Required arguments: input-edgelist output-Graphviz")
+
+G = pgv.AGraph(directed = True)
+
+with open(sys.argv[1], newline = '') as f:
+    reader = csv.reader(f, delimiter = " ")
+    for row in reader: G.add_edge(*row)
+
+G.write(sys.argv[2])

Trajectories/edgelist_test.csv

+1 2
+2 3
+2 4
+3 4
+4 5
+4 9
+5 6
+6 7
+7 8
+9 10
+9 13
+10 11
+11 12
+12 16
+13 14
+14 15
+15 16
+16 26
+26 27
+27 17
+27 20
+17 18
+18 19
+20 21
+21 22
+23 24
+23 25
+25 3

Trajectories/extract_component.py

+import graph_tool
+import csv
+from graph_tool import topology
+
+g = graph_tool.load_graph_from_csv("edgelist.csv", directed = True,
+                                   csv_options = {'delimiter': ' ',
+                                                  'skipinitialspace': True})
+g.set_directed(False)
+
+for v in g.vertices():
+    if g.vp.name[v] == "109114663": break
+
+label = topology.label_out_component(g, v)
+g.set_vertex_filter(label)
+g.purge_vertices()
+g.set_vertex_filter(None)
+g.set_directed(True)
+
+with open("component.csv", "w", newline = "") as f:
+    writer = csv.writer(f, lineterminator = "\n", delimiter = " ")
+
+    for e in g.edges():
+        v_source = e.source()
+        v_target = e.target()
+        writer.writerow([g.vp.name[v_source], g.vp.name[v_target]])