Folium Heat Mapで凡例/グラデーションを追加するには？

Question

私は、Foliumを使用してヒートマップを作成しています。私のデータは3つの列が1つのカテゴリ、緯度と経度を含んでいます。緯度経度はA、B、Cの3つのカテゴリに分類されます。 私は麻痺を使ってヒートマップをプロットすることができますが、ポイント間の色差を示す凡例を追加する必要があります。ポイントをマークする必要がありますそのカテゴリに基づいて3つの異なる色に変換します。

参考用にサンプルコードを添付しています。助けていただければ幸いです。

ありがとうございます！

from folium import plugins 
from folium.plugins import HeatMap 
from folium.plugins import MarkerCluster 
import pandas as pd 

map = folium.Map(location=[lat, long],zoom_start =12) 
data = pd.read_csv(filename) 
# List comprehension to make out list of lists 
heat_data = [[row['LAT'],row['LONG'],] for index, row in data.iterrows()] 
# Plot it on the map 
HeatMap(heat_data).add_to(map) 
# Display the map 
map 
map.save('C:\Temp\map2.html') 

Answer 1

あなたはカラーマップが

steps = 20 
color_map=cm.linear.PuBu.scale(0,1).to_step(steps) 

gradient_map=defaultdict(dict) 
for i in range(steps): 
    gradient_map[1/steps*i] = color_map.rgb_hex_str(1/steps*i) 

値で辞書を作成し、ヒートマップのための勾配としてそれを使用する必要があります。

HeatMap(heat_data, gradient = gradient_map)