pyecharts带你领略动态轨迹图的风骚

文章的开始先打波广告，小文的公众号也开始跟大家见面了，喜欢的我就来'小文的数据之旅'找我玩吧！后台回复pyecharts可获得本文使用的数据源。

ok，进入今天的主题！最近很多朋友问起pyecharts，尤其是地理坐标图的制作，都说被其图形之美给吸引到了。刚好今天也有同事问起来，那么今天就以pyecharts的动态地理轨迹图为例，说说该怎么使用pyecharts。

import pandas as pd     
import random
from pyecharts import GeoLines, Style    #地理轨迹图的类就是Geolines

在同事那里拿了一份国内部分城市的经纬度数据，首先用pandas读取数据，看看数据长什么样子，我们发现数据由一列组成，基本结构为'地名':['经度','纬度'],那么接下来我们将各地方的地名，经度，纬度提取出来。

#读取数据
data = pd.read_excel('./Desktop/data.xlsx',header = None,names = ['name'])
print(data.head())
               name
0      '上海': [121.4648,31.2891],
1       '东莞': [113.8953,22.901],
2      '东营': [118.7073,37.5513],
3       '中山': [113.4229,22.478],
4      '临汾': [111.4783,36.1615],

#写个遍历，把各地方的地名，经度，纬度提取出来，并存取在DataFrame中
city_list = []
lad_list = []
long_list = []
for i in data['name']:
    s = i.strip().split(':')         #去除前后空格，并以":"为分隔符分裂字符
    city = s[0][1:-1]                #取分裂后字符的第一个，得到地名
    lad = s[1].split(',')[0][2:]     #取分裂后字符的第二个，继续以','为分隔符分裂字符
    long = s[1].split(',')[1][:-2]
    city_list.append(city)
    lad_list.append(lad)
    long_list.append(long)
result = pd.DataFrame({'地点': city_list, '经度': lad_list, '纬度': long_list})

从result数据框我们看到，一共由114行，3列数据组成，114行太多了，画到地图上会很乱，于是我们用random包随机抽样20个出来做实验；

在抽样之前，我们需要将数据整理成geolines要求的格式，格式为[('始点','终点')]；

另外因为我们要自定义各城市的经纬度（担心部分城市在地图显示不出来）,pyecharts里面城市经纬度的格式为{'城市':['经度','纬度']},因此我们还需要组装一下我们数据。

#以东莞为始点，其他各个城市为终点，整理数据
plotting = result[result['地点'] != '东莞']['地点'].apply(lambda x : ('东莞',x))

#自定义各城市的经纬度
geo_cities_coords = {result.iloc[i]['地点']:[result.iloc[i]['经度'],result.iloc[i]['纬度']] for i in range(len(result))}

#随机抽样20个城市组合
plotting_data = random.sample(list(plotting),20)

数据已经准备好了，接下来就套路来了，先设定画布的格式，然后将geolines作为类Geolines的实例，接着设置geolines的参数，最后展示成果！

#设置画布的格式
style = Style(title_pos="center", 
              width=1000, 
              height=800)

#部分地理轨迹图的格式
style_geolines = style.add(is_label_show=True,
                      line_curve=0.3,             #轨迹线的弯曲度，0-1
                      line_opacity=0.6,           #轨迹线的透明度，0-1
                      geo_effect_symbol='plane',  #特效的图形，有circle,plane,pin等等
                      geo_effect_symbolsize=10,   #特效图形的大小
                      geo_effect_color='#7FFFD4', #特效的颜色
                      geo_effect_traillength=0.1, #特效图形的拖尾效果，0-1
                      label_color=['#FFA500', '#FFF68F'],#轨迹线的颜色，标签点的颜色，
                      border_color='#97FFFF',     #边界的颜色
                      geo_normal_color='#36648B', #地图的颜色
                      label_formatter='{b}',      #标签格式
                      legend_pos = 'left')

#作图
geolines = GeoLines('小文的pyechart出行轨迹图', **style.init_style)
geolines.add('从东莞出发',
             plotting_data,
             maptype='china',    #地图的类型，可以是省的地方，如'广东',也可以是地市，如'东莞'等等
             geo_cities_coords=geo_cities_coords,
             **style_geolines)

#发布，得到图形的html文件
geolines.render()                       

做到这里地理轨迹图已经完成了，那么假如我们想画一个既有出发又有回来的轨迹图怎么画呢？其实很简单，就在上面的基础上，再增加一个add就可以了，如下：

geolines.add('回到东莞',
             [('海口','东莞'),('拉萨','东莞'),('贵州','东莞'),('兰州','东莞')],
             maptype='china',  
             geo_cities_coords=geo_cities_coords,
             **style_geolines)

好了，我们看到了有几架小飞机回到东莞了，分别是我们设置的拉萨，贵州，海口和兰州，这样看似乎有点乱，那么我们可以设置一个参数，使得出发和回程分开，先设置is_legend_show = True,再设legend_selectdmode = 'single',点击不同的图例就可以显示了，如下：

style_geolines = style.add(is_label_show=True,
                      line_curve=0.3,             
                      line_opacity=0.6,           
                      geo_effect_symbol='plane', 
                      geo_effect_symbolsize=10,   
                      geo_effect_color='#7FFFD4',
                      geo_effect_traillength=0.1,
                      label_color=['#FFA500', '#FFF68F'],
                      border_color='#97FFFF',  
                      geo_normal_color='#36648B', 
                      label_formatter="{b}", 
                      is_legend_show=True,
                      legend_pos = 'left',
                      legend_selectdmode = 'single')   #单例模式
geolines.add('从东莞出发',          #图例1名称
             plotting_data,
             maptype='china',    
             geo_cities_coords=geo_cities_coords,
             **style_geolines)
geolines.add('回到东莞',            #图例2名称
             [('海口','东莞'),('拉萨','东莞'),('贵州','东莞'),('兰州','东莞')],
             maptype='china',  
             geo_cities_coords=geo_cities_coords,
             **style_geolines)