根据经纬度生成Geohash

#---zhangpengxu---
#-*- coding: utf-8 -*-

from pygeohash import encode, decode
import plotly
import plotly.plotly as pyf
import plotly.graph_objs as go
import numpy as np
import pandas as pd
import math
from matplotlib.path import Path
import numpy as np
import plotly.offline as of
import plotly.graph_objs as go
import plotly.plotly as py
import numpy as np
import pandas as pd
import folium
import webbrowser
from folium.plugins import HeatMap
import datetime
import time
import pymysql.cursors
import decimal
# 创建一个游标


def mysql():
    conn = pymysql.connect(
        host='localhost',
        port=3306,
        user='root',
        passwd='xu19931026',
        db='cd_taxi',
        charset='utf8'
    )
    if conn:
        print("连接成功!")
    cursor = conn.cursor()  # 获取游标
    sql = "select CAST(BeginLongitude as CHAR(11)) as BeginLongitude,CAST(BeginLatitude as CHAR(10)) as BeginLatitude FROM Order_Data where FROM_UNIXTIME(BeginTime) <'2016-11-01 00:05:00' "  # sql语句
    cursor.execute(sql)
    result=cursor.fetchall()
    df=list(result)  #将元组转换为列表
    lon = []
    lat = []
    for point in df:
        lon.append(float(point[0]))
        lat.append(float(point[1]))
    num=len(lon) #num为数组大小
    data = [ [lon[i],lat[i]] for i in range(num) ]    #将数据制作成[lats,lons,weights]的形式
    print('**************************************')
    print('数据库处理数据完毕')
    return lon,lat,data
def get_lonandlat(geo):
    lon, lat = decode(geo)
    return lon, lat
def get_geohash(lon, lat):
    # 获取大致分区：
    #生成Geohash
    geo = encode(lon, lat)
    return geo
def get_geolist(lon,lat):  #根据MySQL出的经纬度生成Geohash
    p = []
    for i in range(len(lon)):
        result = get_geohash(lon[i],lat[i])
        p.append(result)
    return p
def new_geohash(lon,lat):
    z = list(zip(lon,lat))
   #new_total_lonlat是list列表[[103.0,30.5],[103.2,31.1].......]
    return z
# get_dict得到分块信息字典，键为各个中心点坐标，值为各个中心点形成的矩形区域的四个顶点。
def get_dict(central_points, lons_jia, lats_jia):
    point_dict = {}
    for central_point in central_points:
        border_points = []
        lons = [central_point[0] +lons_jia, central_point[0] - lons_jia]
        lats = [central_point[1] + lats_jia, central_point[1] - lats_jia]
        for lon in lons:
            for lat in lats:
                border_points.append([lon, lat])
        point_dict[str(central_point)] = border_points
    return point_dict

#*********************************************************
# 画图
def geo_paint(dict, lon, lat,new_geohash):
    #画中心点

    datas = [
        go.Scattermapbox(
            lat=lat,
            lon=lon,
            text=new_geohash,
            mode='markers',
            # hoverinfo='text', #如果hoverinfo包含'text', 并且'hovertext'未设置, 则这些文本元素将显示在悬停标签中https://zhuanlan.zhihu.com/p/87163211?utm_source=qq
            marker=go.scattermapbox.Marker(
                size=5,
                color='red',  # 000045
                opacity=0.8
            )
        )
    ]
    # 遍历每一个中心点，并画出对应矩形范围（一个一个区域画）
    for key in dict.keys():
        borders = dict[key]
        lons = [] #为new_dict[key]例的经度
        lats = []
        for border in borders:  #border 为每个key后对应的四个坐标经纬度
            lons.append(border[0])
            lats.append(border[1])
        lons1 = list(set(lons))
        lats1 = list(set(lats))
        lats1.sort()
        lons1.sort()
        lons = [lons1[0], lons1[0], lons1[1], lons1[1], lons1[0]] #画成一个封闭的四方形 lons经度为102.95小数点后两位
        lats = [lats1[0], lats1[1], lats1[1], lats1[0], lats1[0]]
        # lons = pd.DataFrame(lons)
        # lats = pd.DataFrame(lats)
        datas.append(
            go.Scattermapbox(
                lat=lats,
                lon=lons,
                # text=geo,
                mode='lines',
                # hoverinfo='text',
                marker=go.scattermapbox.Marker(
                    size=3,
                    color='red',
                    opacity=0.8
                )
            )
        )
    # 获取地图
    mapbox_access_token = '''pk.eyJ1IjoiemhhbmdwZW5neHUiLCJhIjoiY2s1ZGwxbXpzMHhjdTNlbnFteTdlcXk5eiJ9.vzYUBRLW8vw7L_H9ybiJYg'''
    layout = go.Layout(
        title="chengdu",
        autosize=True,
        hovermode='closest',
        showlegend=False,
        #暗黑地图

        mapbox=go.layout.Mapbox(
            accesstoken=mapbox_access_token,
            bearing=0,

            center=go.layout.mapbox.Center(
                lat=30.67,  # 成都市纬度
                lon=104.06  # 广州市经度
            ),
            pitch=0,
            zoom=10,
            style='light'
        ),
    )

    fig = go.Figure(data=datas, layout=layout)
    of.plot(fig, filename='Chengdu_geo.html')  # 生成html文件并打开

#*************************************************


if __name__ == "__main__":
    start = time.clock()
    lon,lat,data=mysql()  #返回MySQL出的lon,lat
    end1=time.clock()
    print('mysql处理完毕;')
    print("耗时", end1 - start)
    print('*********************************')
    geohash1=get_geolist(lon,lat)  #据MySQL出的经纬度生成Geohash
    end2 = time.clock()
    print('MySQL出的经纬度生成Geohash处理完毕;')
    print("耗时", end2 - end1)
    print('*********************************')
    new_geohash_lonlat = new_geohash(lon,lat)  #返回前N位的新Geohash和根据新Geohash解析出的新经纬度
    end3 = time.clock()
    print('经度纬度合并得到经纬度列表;')
    print("耗时", end3 - end2)
    print('*********************************')
    dict=get_dict(new_geohash_lonlat,0.001,0.001)
    end4 = time.clock()
    print('得到经纬度前后左右加0.01用于画方块;')
    print("耗时", end4 - end3)
    print('开始画方块;')
    print('*********************************')
    geo_paint(dict, lon, lat,geohash1)
    end5 = time.clock()
    print("已画完一共耗时", end5 - start)
    print('*********************************')