Фолиевые тепловые карты - PullRequest
1 голос
/ 19 апреля 2020

Тепловые карты в Folium

У меня есть информационный фрейм с данными о транзакциях дома (метаданные о доме, местоположении и цене).

Я хотел бы сделать тепловую карту в folium, представляющем цену, а не частоту по умолчанию. Таким образом, области с высокими ценами сделок становятся более разноцветными и наоборот.

Кто-нибудь знает? Моим лучшим предположением было бы представить цену в виде числа строк, например, если бы одна строка была одной ценовой единицей, а дом был продан за 5 единиц, то мне нужно было бы продублировать строку 5 раз. Но это просто не достаточно дружественный к вычислениям (если вообще возможно).

Спасибо!

Вот представление моих данных:

data = pd.DataFrame({"house_id" : [1, 2, 3, 4],
                     "latitude": [55.83, 55.79, 55.86, 55.85],
                     "longitude": [12.05, 12.10, 12.07, 12.09],
                     "price": [2000, 2100, 1500, 4500]})

И вот код, который я использую для создания тепловой карты с помощью Folium, которая к настоящему времени делает "жара" с точки зрения того, как часто местоположение apperars с точки зрения #rows ..

from folium import plugins
from folium.plugins import HeatMap

heat_data = [[row['latitude'],row['longitude']] for index, row in data.iterrows()]
map = folium.Map(location=[55.838913, 12.055415], zoom_start = 13)

# Plot it on the map
HeatMap(heat_data).add_to(map)

# Display the map
map
...