In [1]:
import pandas as pd
In [3]:
df = pd.read_csv('/Users/chengjun/bigdata/soda-2017-sample/2017/Mobike_SODA_Sample/shanghai_sample.csv')
In [4]:
df.head()
Out[4]:
In [5]:
len(df)
Out[5]:
In [21]:
df["userid"].value_counts()
Out[21]:
In [15]:
starts = []
ends = []
edges = []
for i in df.index:
starts.append((df.start_location_y[i], df.start_location_x[i]))
ends.append((df.end_location_y[i], df.end_location_x[i]))
edges.append([(df.start_location_y[i], df.start_location_x[i]), (df.end_location_y[i], df.end_location_x[i])])
In [34]:
df5054 = df[df['userid']==20]
edges5054 = []
for i in df5054.index:
edges5054.append([(df5054.start_location_y[i], df5054.start_location_x[i]), \
(df5054.end_location_y[i], df5054.end_location_x[i])])
In [13]:
starts[:5]
Out[13]:
In [6]:
import folium, jinja2, vincent
from IPython.display import IFrame
from IPython.core.display import HTML
print(folium.__file__)
print(folium.__version__)
In [7]:
shanghai = folium.Map(location=(31.306,121.410), zoom_start=10)
In [8]:
shanghai
Out[8]:
In [36]:
from folium import plugins
shanghai = folium.Map(location=(31.306,121.410), zoom_start=10)
shanghai.add_children(plugins.HeatMap(starts[:1000]))
shanghai
Out[36]:
In [19]:
shanghai = folium.Map(location=(31.306,121.410), zoom_start=10)
for loc in edges[:1000]:
line = folium.PolyLine(locations=loc)
shanghai.add_children(line)
In [20]:
shanghai
Out[20]:
In [35]:
shanghai5054 = folium.Map(location=(31.306,121.410), zoom_start=10)
for loc in edges5054:
line = folium.PolyLine(locations=loc)
shanghai5054.add_children(line)
shanghai5054
Out[35]:
In [ ]: