-
Notifications
You must be signed in to change notification settings - Fork 2
/
zupdategexf.py
69 lines (40 loc) · 1.38 KB
/
zupdategexf.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import glob
import numpy as np
import networkx as nx
import os
os.system('rm centrality/circle_* ')
files = glob.glob('centrality/*.gexf')
try:
from xml.dom import minidom
doc = minidom.parse('centrality/template.svg') # parseString also exists
x = [c.getAttribute('cx') for c in doc.getElementsByTagName('circle')]
y = [c.getAttribute('cy') for c in doc.getElementsByTagName('circle')]
id = [c.getAttribute('class').replace('id_','') for c in doc.getElementsByTagName('circle')]
doc.unlink()
print 'read template'
for f in files:
print f
G = nx.read_gexf(f)
attrs = {}
for i in range(len(x)):
attrs[id[i]] = {'x':float(x[i]),'y':float(y[i])}
nx.set_node_attributes(G, attrs)
nx.write_gexf(G,'centrality/template_'+f.split('/')[-1])
except:
f=files[0]
G = nx.read_gexf(f)
sorteddeg = sorted(dict(G.degree()).items(), key=lambda kv: kv[1],reverse=True)
sdl = len(sorteddeg)
fract = 2*np.pi/sdl
idx = []
groups = 60
scale = 1000
for i in range(groups):
idx.extend(range(i,sdl,groups))
angle = 0
attrs = {}
for i in idx:
angle+=fract
attrs[sorteddeg[i][0]] = {'x':float(scale*(np.cos(angle))),'y':float(scale*(np.sin(angle)))}
nx.set_node_attributes(G, attrs)
nx.write_gexf(G,'centrality/circle_'+f.split('/')[-1])