def pickEndCountry(metric):
endCountries = COS.nordicCountries() + COS.westernEuropeCountries()
endCoLabels = []
endCoWeight = []
for vertex in graph.vs:
if vertex['label'] in endCountries:
# print vertex
endCoLabels.append(vertex['ID'])
# endCoWeight.append( vertex[metric] )
# print len(endCoLabels)
# print len(endCoWeight)
endCoWeight = [ i/sum(endCoWeight) for i in endCoWeight ]
return np.random.choice(endCoLabels)#, p=endCoWeight)
if graph.vs[vertexNumber]['label'] == 'Syria':
graph.vs[vertexNumber]['color'] = 'rgba(78,0,18, '+str(vertexOpacity)+')'
graph.vs[vertexNumber]['NumRefs'] = 3000000
elif graph.vs[vertexNumber]['label'] in COS.africaCountries():
graph.vs[vertexNumber]['color'] = 'rgba(156, 52, 76, '+str(vertexOpacity)+')'
graph.vs[vertexNumber]['NumRefs'] = 5000
elif graph.vs[vertexNumber]['label'] in COS.middleEastCountries():
graph.vs[vertexNumber]['color'] = 'rgba(170, 95, 57, '+str(vertexOpacity)+')'
graph.vs[vertexNumber]['NumRefs'] = 5000
elif graph.vs[vertexNumber]['label'] in COS.centralEuropeCountries():
graph.vs[vertexNumber]['color'] = 'rgba(0,0,100, '+str(vertexOpacity)+')'
graph.vs[vertexNumber]['NumRefs'] = 6000
elif graph.vs[vertexNumber]['label'] in COS.southernEuropeCountries():
graph.vs[vertexNumber]['color'] = 'rgba(38, 113, 88, '+str(vertexOpacity)+')'
graph.vs[vertexNumber]['NumRefs'] = 3000
elif graph.vs[vertexNumber]['label'] in COS.westernEuropeCountries():
graph.vs[vertexNumber]['color'] = 'rgba(87,49,50, '+str(vertexOpacity)+')'
graph.vs[vertexNumber]['NumRefs'] = 3000
elif graph.vs[vertexNumber]['label'] in COS.nordicCountries():
graph.vs[vertexNumber]['color'] = 'rgba(0,57,38, '+str(vertexOpacity)+')'
graph.vs[vertexNumber]['NumRefs'] = 3000
else:
print 'WARNING: No color specified for: ',graph.vs[vertexNumber]['label']
## Place edge properties on edges
countryDistances = COS.distanceBetweenCountries()
for edge in graph.es: