In [2]:

    

import networkx as nx
import cPickle as pik
import datetime

data = pik.load(open("/home/raj/projects/atari/data/mailElist.pik","rb"))
tmp=[x[0:2] for x in data]
nodes=[val for sublist in tmp for val in sublist]
print len(nodes)
nodes=list(set(nodes))
len(nodes)


    

    




11774






    
Out[2]:





813

In [3]:

    

type(data[160][2])


    

    
Out[3]:





datetime.datetime

In [4]:

    

g = nx.DiGraph()
for x in nodes:
    g.add_node(nodes.index(x),name=x)
    
for x in data:
    if type(x[2])==datetime.datetime:
        s = nodes.index(x[0])
        t = nodes.index(x[1])
        e = g.add_edge(int(s), int(t),sent=x[2].strftime("%Y-%m-%d"))


    

In [5]:

    

import matplotlib.pyplot as plt
nx.draw(g)
plt.show()


    

In [77]:

    

nx.write_graphml(g, "vaxemail.graphml")


    

In [ ]: