scripts_igraph.py

from igraph import Graph
import numpy as np

graph = np.loadtxt('data/I210/graph.csv', delimiter=',')
od = np.loadtxt('data/I210/od.csv', delimiter=',')
# vertices = [i for i in range(7)]
# edges = [(0, 2), (0, 1), (0, 3), (1, 0), (1, 2), (1, 3), (2, 0), (2, 1),
#          (2, 3), (3, 0), (3, 1), (3, 2), (2, 4), (4, 5), (4, 6), (5, 4),
#          (5, 6), (6, 4), (6, 5)]
# g = Graph(vertex_attrs={"label":vertices}, edges=edges, directed=True)


vertices = range(int(np.max(graph[:, 1:3])) + 1)
# print vertices
edges = graph[:, 1:3].astype(int).tolist()
weights = graph[:, 3].tolist()
# print edges
# print weights
g = Graph(vertex_attrs={"label": vertices}, edges=edges, directed=True)
out = g.get_shortest_paths(18, to=19, weights=weights, output="vpath")
print out
out = g.get_shortest_paths(18, to=[11, 19], weights=weights, output="epath")
print out
print len(g.es)
# print [e for e in g.es]

g.es["weight"] = weights
out = g.get_shortest_paths(18, to=19, weights="weight", output="vpath")
print out
out = g.get_shortest_paths(18, to=[11, 19], weights="weight", output="epath")
print out

vertices = range(int(np.max(graph[:, 1:3])) + 2)
print vertices
print edges
edges.append([20, 19])
weights.append(1.0)
g = Graph(vertex_attrs={"label": vertices}, edges=edges, directed=True)
g.es["weight"] = weights
out = g.get_shortest_paths(
    18, to=[11, 19, 20], weights=weights, output="vpath")
print out