python热力图的原理实现

import seaborn as sns

%matplotlib inline

sns.set(font_scale=1.5)

import gmplot             # plot the locations on google map

import numpy as np        # linear algebra

import pandas as pd       # data processing, CSV file I/O (e.g. pd.read_csv())

import matplotlib.pyplot as plt  # data visualization

import seaborn as sns       # data visualization

df = pd.read_csv("data.csv")

df = pd.DataFrame(df)

df_td = pd.read_csv("datacopy.csv")

df_td = pd.DataFrame(df_td)

# print df.dtypes

print (df.shape)

print (df_td.shape)

def plot_heat_map(data, number):

    latitude_array = data['INTPTLAT'].values

    latitude_list = latitude_array.tolist()

    print(latitude_list[0])

    Longitude_array = data['INTPTLONG'].values

    longitude_list = Longitude_array.tolist()

    print(longitude_list[0])

    # Initialize the map to the first location in the list

    gmap = gmplot.GoogleMapPlotter(latitude_list[0], longitude_list[0], 10)

    # gmap.scatter(latitude_list, longitude_list, edge_width=10)

    gmap.heatmap(latitude_list, longitude_list)

    # Write the map in an HTML file

    # gmap.draw('Paths_map.html')

    gmap.draw('{}_Paths_map.html'.format(number))

plot_heat_map(df,'4')