Read edgelist with attributes as igraph Graph and plot it with nodes labeled

read_igraph.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

from igraph import *


def parse_string(x, max_len=100):
    out = 'NONE'
    try:
        out = str(x.encode('ascii'))[:max_len]
    except:
        pass
    return out


def read_file(
    filename,
    directed=True,
    etype_names=None,
    header=False,
    ):

    edgelist = []
    etypes = []
    names = UniqueIdGenerator()
    header_info = []
    with open(filename) as fp:
        data = fp.readlines()[0].split('\r')
        for line in data:
            line = line.strip().split(',')
			if header:
                header_info = line
                header = False
                continue
            n1 = names[line[0]]
            n2 = names[line[1]]
            edgelist.append((n1, n2))
            if etype_names is not None and len(line) \
                == len(etype_names) + 2:
                etypes.append(tuple(line[2:]))
    g = Graph(edgelist, directed=directed)
    g.vs['name'] = [names.reverse_dict()[vid.index] for vid in g.vs]
    g.vs['label'] = [parse_string(v) for v in g.vs['name']]
    if etype_names is not None:
        for (k, v) in zip(etype_names, zip(*etypes)):
            g.es[k] = v
    return g


etype_names = ['Edge Type', 'weight', 'directed']
g = read_file('test.csv', etype_names=etype_names, directed=False,
              header=True)
print g.summary()

## Plot the graph

layout = g.layout('reingold_tilford')
plot(g, layout=layout)

test.csv