In [37]:

    

import networkx as nx
import pandas as pd
from matplotlib import pyplot as plt
%matplotlib inline


    

In [68]:

    

a = pd.read_csv('test/results/lsi-results-stopwords-spanish-3.net')
a = a.groupby('from_name').head(2).reset_index(drop=True)
print a.head()


    

    




                          from_name       sim  \
0  12_dibujos_de_jose_maria_velasco  1.000000   
1  12_dibujos_de_jose_maria_velasco  0.647445   
2    20_dibujos_mexicanos_de_maroto  1.000000   
3    20_dibujos_mexicanos_de_maroto  0.223288   
4                            vitral  1.000000   

                            to_name  from  to  
0  12_dibujos_de_jose_maria_velasco     0   0  
1                          zurbaran     0   4  
2    20_dibujos_mexicanos_de_maroto     1   1  
3                            vitral     1   2  
4                            vitral     2   2  

In [69]:

    

g = nx.DiGraph()
g.add_nodes_from(a['from_name'])
tups = []
for i in range(len(a)):
    tups.append(tuple(a[['from_name','to_name']].iloc[i,:]))
g.add_edges_from(tups, )


    

In [70]:

    

[i for i in dir(nx) if 'layout' in i]


    

    
Out[70]:





['circular_layout',
 'fruchterman_reingold_layout',
 'graphviz_layout',
 'layout',
 'pydot_layout',
 'pygraphviz_layout',
 'random_layout',
 'shell_layout',
 'spectral_layout',
 'spring_layout']

In [33]:

    

pos = nx.layout.fruchterman_reingold_layout(g)
nx.draw(g, pos, node_color='#00AAFF', edge_color='#0088CC')
plt.savefig('test/results/lsi-graph-stopwords-spanish-3.png')


    

In [ ]: