如何通过自定义半径创建地理热图

Question

我想使用叶片在地图上创建可视化效果。在地图中，我想观察与构建热图的特定地理位置相关的项目有多少。下面是我正在使用的代码。

import pandas as pd
import folium
from folium import plugins

data = [[41.895278,12.482222,2873494.0,20.243001,20414,7.104243],
        [41.883850,12.333330,3916.0,0.835251,4,1.021450],
        [41.854241,12.567000,22263.0,1.132390,35,1.572115],
        [41.902147,12.590388,19505.0,0.839181,37,1.896950],
        [41.994240,12.48520,16239.0,1.383981,25,1.539504]]

df = pd.DataFrame(columns=['latitude','longitude','population','radius','count','normalized'],data=data)

middle_lat = df['latitude'].median()
middle_lon = df['longitude'].median()
m = folium.Map(location=[middle_lat, middle_lon],tiles = "Stamen Terrain",zoom_start=11)

# convert to (n, 2) nd-array format for heatmap
points = df[['latitude', 'longitude', 'normalized']].dropna().values

# plot heatmap
plugins.HeatMap(points, radius=15).add_to(m)
m.save(outfile='map.html')

这里是结果

在此地图中，每个点都具有相同的半径。插入后，我想创建一个热图，其点半径与其所属的城市之一成正比。我已经尝试过在列表中传递半径，但是它不起作用，以及通过for循环传递值。

有什么主意吗？

Answer 1

您需要一个接一个地添加一个点。因此，您可以指定每个点的半径。像这样：

import random
import numpy

pointArrays = numpy.split(points, len(points))
radii = [5, 10, 15, 20, 25]

for point, radius in zip(pointArrays, radii):
    plugins.HeatMap(point, radius=radius).add_to(m)

m.save(outfile='map.html')

在这里您可以看到，每个点都有不同的大小。