好事成双，图层锁定轨迹追踪均完成

# coding=utf-8

import csv

import numpy as np
import folium
from folium import plugins
import csv
import requests
import json
import pandas as pd
import re
import time

import webbrowser




def PlotLineOnMap():
    Lat = []
    Lon = []
    Category = []
    Date_time = []
    print(type(Lat))
    fn = \'result.csv\'
    with open(fn) as csvFile:
        csvDictReader = csv.DictReader(csvFile)
        for row in csvDictReader:
            print(row[\'lon\'], row[\'lat\'])
            Lat.insert(0,float(row[\'lat\']))
            Lon.insert(0,float(row[\'lon\']))
            Category.insert(0,str(row[\'category\']))
            Date_time.insert(0,str(row[\'date_time\']))
            print(Lat)


    # 给出的坐标系为GCJ-02，如果需要测试google地图，需要进行坐标转换

    tri = np.array(list(zip(Lat, Lon,Category,Date_time)))

    san_map = folium.Map(
        location=[36.641254, 117.068689],
        zoom_start=16,
        # 高德街道图
        tiles=\'http://webrd02.is.autonavi.com/appmaptile?lang=zh_cn&size=1&scale=1&style=7&x={x}&y={y}&z={z}\',
        # tiles=\'http://webst02.is.autonavi.com/appmaptile?style=6&x={x}&y={y}&z={z}\', # 高德卫星图
        attr=\'default\')

    layer_0329 = folium.FeatureGroup(name=\"0329\")
    layer_0330 = folium.FeatureGroup(name=\"0330\")
    layer_0331 = folium.FeatureGroup(name=\"0331\")
    layer_0401 = folium.FeatureGroup(name=\"0401\")


    #folium.PolyLine(tri, color=\'#3388ff\').add_to(san_map)
  #  marker_cluster = plugins.MarkerCluster().add_to(san_map)
    for lat, lon,category,date_time in zip(Lat, Lon,Category,Date_time):
        print([category])
        print(type(category))
        print(category)
        print([date_time])

        if( date_time == \'0329\' ):
            if (category == \'确诊者住址\'):
                folium.Marker(location=[lat, lon], popup=\'Mt. Hood Meadows\',
                              icon=folium.Icon(icon=\'cloud\', color=\"red\")).add_to(layer_0329)
            elif (category == \'确诊者轨迹\'):
                folium.Marker(location=[lat, lon], popup=\'Mt. Hood Meadows\',
                              icon=folium.Icon(icon=\'cloud\', color=\'orange\')).add_to(layer_0329)









    san_map.add_child(layer_0329)
    san_map.add_child(layer_0330)
    san_map.add_child(layer_0331)
    san_map.add_child(layer_0401)





    san_map.add_child(folium.map.LayerControl())

    san_map.add_child(folium.LatLngPopup())
    san_map.save(\'test.html\')


def main():
    num = 0
    y = []
    \'\'\'
    在url里key后面的******换为高德开放平台自己申请的key
    \'\'\'
    with open(\"地址.csv\", \'r\') as f:  # 写入将要转换的地址的文件路径,此处为默认文件路径（要先将文件提前导入）（注意是csv格式文件）
        r = csv.reader(f, delimiter=\',\')
        for row in r:
            print(row[0])
            print(row[1])
            url = \"http://restapi.amap.com/v3/geocode/geo?key=831894c7f8fe1d2bedb982a040536650&address=\" + row[2]
            dat = {
                \'count\': \"1\",
            }
            r = requests.post(url, data=json.dumps(dat))
            s = r.json()
            try:
                b = s[\'geocodes\']
                text = str(b)
                print(text)
            except:
                text = \'none\'

            # 省份
            try:
                pat1 = \"\'province\': \'(.*?)\',\"
                province = re.compile(pat1).findall(text)[0]
                print(\"省份：\", province)
            except:
                province = \'none\'
                print(\"省份：\", province)

            # 城市
            try:
                pat2 = \"\'city\': \'(.*?)\',\"
                city = re.compile(pat2).findall(text)[0]
                print(\"城市：\", city)
            except:
                city = \'none\'
                print(\"城市：\", city)

            # 区划
            try:
                pat3 = \"\'district\': \'(.*?)\',\"
                district = re.compile(pat3).findall(text)[0]
                print(\"区划：\", district)
            except:
                district = \'none\'
                print(\"区划：\", district)

            # 经纬度
            try:
                pat4 = \"\'location\': \'(.*?)\',\"
                res3 = re.compile(pat4).findall(text)[0]
                # print(res3)
                lon_lat = res3.split(\',\')
                lon = float(lon_lat[0])
                lat = float(lon_lat[1])
                print(\"经度：\", lon)
                print(\"纬度：\", lat)
            except:
                lon = \'none\'
                lat = \'none\'
                print(\"经度：\", lon)
                print(\"纬度：\", lat)

            num += 1
            print(\"第\" + str(num) + \"条地址转换成功\")
            print(\'**************************************\')
            date_time =row[0]
            y.append([num,str(row[0]),row[1],row[2], province, city, district, lon, lat])
            result = pd.DataFrame(y)
            result.columns = [\'num\', \'date_time\',\'category\',\'address\', \'province\', \'city\', \'district\', \'lon\', \'lat\']
            result.to_csv(\'result.csv\', encoding=\'gbk\', index=False)
        print(\"全部地址转换成功\")

    time.sleep(5)  #暂停10秒
    PlotLineOnMap()

    webbrowser.open(\'test.html\')



if __name__ == \'__main__\':
    main()