import folium, jinja2, vincent
from IPython.display import IFrame
from IPython.core.display import HTML
print(folium.__file__)
print(folium.__version__)
/Users/chengjun/anaconda/lib/python2.7/site-packages/folium-0.3.0.dev0-py2.7.egg/folium/__init__.pyc 0.3.0.dev
import pickle
node_pkl = open('/Users/chengjun/GitHub/cjc2016/data/nodeLoc30.pkl', 'rb')
nodeLoc = pickle.load(node_pkl)
node_pkl.close()
edge_pkl = open('/Users/chengjun/GitHub/cjc2016/data/edgeLoc30.pkl', 'rb')
edgeLoc = pickle.load(edge_pkl)
edge_pkl.close()
for i in edgeLoc: print i
12296 12181 12064 12196 12202 12340 12347 12092 12100 12230 12407 12238 12112 12241 12370 12243 12372 12245 12122 12124 12261 12136 12010 12014 12274 12023 12156 12285 12158 12287
beijing = folium.Map(location=(39.90403,116.407526), zoom_start=10)
from folium.plugins import MarkerCluster
locations = [j for n in nodeLoc.values() for j in n.values()]
popups = ['{}'.format(loc) for loc in locations]
beijing.add_children(MarkerCluster(locations=locations, popups=popups))
from folium import plugins
beijing.add_children(plugins.HeatMap(locations))
import matplotlib
color_num = 30
cmap = matplotlib.cm.get_cmap('jet',color_num)
colors = [matplotlib.colors.rgb2hex(cmap(i)) for i in range(color_num)]
#colors = ['blue','orange', 'red']
for k, community in enumerate(edgeLoc):
locations = edgeLoc[community]
for loc in locations:
line = folium.PolyLine(locations=loc, color=colors[k])
beijing.add_children(line)
beijing
beijing.save('/Users/chengjun/GitHub/cjc2016/vis/beijing_mobile_30_line.html')
beijing.save('/Users/chengjun/GitHub/cjc2016/vis/beijing_mobile_30_heat.html')