SYNC COMMIT
This commit is contained in:
Sebastian Lenzlinger
parent
7ae7d486f1
commit
758bfa45f3
@ -1,11 +1,12 @@
|
||||
import pandas as pd
|
||||
import geopandas as gpd
|
||||
import os
|
||||
import colorsys
|
||||
import folium
|
||||
from folium import plugins
|
||||
import logging
|
||||
|
||||
from folium.plugins import HeatMap
|
||||
from matplotlib import pyplot as plt
|
||||
|
||||
from db_connector import RemoteDB
|
||||
import shapely
|
||||
@ -91,6 +92,8 @@ def create_heat_map_toggle(folium_map):
|
||||
|
||||
add_heat_year_toggle(heat_gdf, folium_map)
|
||||
|
||||
add_bike_heat_toggle(folium_map)
|
||||
add_ped_heat_map(folium_map)
|
||||
# Add signald speeds data
|
||||
add_signaled_speeds(folium_map)
|
||||
|
||||
@ -113,11 +116,12 @@ def add_bike_heat_map_time(folium_map):
|
||||
index = [str(element) for element in index]
|
||||
index = [AccidentType + element for element in index]
|
||||
# plot heat map
|
||||
gradient = generate_hue_gradient(0.6, 5)
|
||||
hm = plugins.HeatMapWithTime(heat_data,
|
||||
auto_play=False,
|
||||
max_opacity=0.8,
|
||||
max_opacity=1,
|
||||
gradient=gradient,
|
||||
min_opacity=0.3,
|
||||
min_opacity=0.5,
|
||||
radius=9,
|
||||
use_local_extrema=False,
|
||||
blur=1,
|
||||
@ -133,7 +137,7 @@ def add_pedestrian_heat_map_time(folium_map):
|
||||
heat_df = gpd.GeoDataFrame(pedestrian_heat_view_data, columns=['latitude', 'longitude', 'year'])
|
||||
|
||||
assert not heat_df.empty, f" Heat Dataframe is empty: {heat_df.head(5)}"
|
||||
heat_data = [[[row['latitude'], row['longitude'], 0.1] for index, row in heat_df[heat_df['year'] == i].iterrows()] for
|
||||
heat_data = [[[row['latitude'], row['longitude'], 0.5] for index, row in heat_df[heat_df['year'] == i].iterrows()] for
|
||||
i in range(2011, 2023)]
|
||||
logger.debug(f"First element of PED heat data: {heat_data[0]}")
|
||||
index = [2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022]
|
||||
@ -142,11 +146,12 @@ def add_pedestrian_heat_map_time(folium_map):
|
||||
index = [AccidentType + element for element in index]
|
||||
#gradient =
|
||||
# plot heat map
|
||||
gradient = generate_hue_gradient(0.2, 5)
|
||||
hm = plugins.HeatMapWithTime(heat_data,
|
||||
auto_play=False,
|
||||
max_opacity=0.8,
|
||||
max_opacity=1,
|
||||
gradient=gradient,
|
||||
min_opacity=0.3,
|
||||
min_opacity=0.5,
|
||||
radius=9,
|
||||
use_local_extrema=False,
|
||||
blur=1,
|
||||
@ -156,7 +161,7 @@ def add_pedestrian_heat_map_time(folium_map):
|
||||
|
||||
|
||||
def add_heat_map_time(heat_df, folium_map):
|
||||
heat_data = [[[row['latitude'], row['longitude'], 0.1] for index, row in heat_df[heat_df['year'] == i].iterrows()] for
|
||||
heat_data = [[[row['latitude'], row['longitude'], 0.5] for index, row in heat_df[heat_df['year'] == i].iterrows()] for
|
||||
i in range(2011, 2023)]
|
||||
index = [2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022]
|
||||
# create heat map
|
||||
@ -170,7 +175,6 @@ def add_heat_map_time(heat_df, folium_map):
|
||||
use_local_extrema=False,
|
||||
blur=1,
|
||||
index=index,
|
||||
overlay=False,
|
||||
name="Accident Heatmap ALL")
|
||||
hm.add_to(folium_map)
|
||||
|
||||
@ -199,7 +203,7 @@ def add_signaled_speeds(folium_map):
|
||||
).add_to(folium_map)
|
||||
|
||||
|
||||
def add_heat_year_toggle(heat_gdf, folium_map):
|
||||
def add_heat_year_toggle(heat_gdf, folium_map, name="All"):
|
||||
index = [2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022]
|
||||
# plot heat map
|
||||
for year in index:
|
||||
@ -213,11 +217,26 @@ def add_heat_year_toggle(heat_gdf, folium_map):
|
||||
max_opacity=0.8,
|
||||
blur=10,
|
||||
show=False,
|
||||
name=f'Accidents in {year}'
|
||||
name=f'Accidents involving {name} in {year}'
|
||||
)
|
||||
|
||||
heatmap_layer.add_to(folium_map)
|
||||
|
||||
|
||||
def add_bike_heat_toggle(folium_map):
|
||||
bike_heat_view_data = get_view('bikeheat', 'latitude, longitude, year')
|
||||
heat_gdf = gpd.GeoDataFrame(bike_heat_view_data, columns=['latitude', 'longitude', 'year'])
|
||||
add_heat_year_toggle(heat_gdf, folium_map, name="motorcycles")
|
||||
|
||||
|
||||
def add_ped_heat_map(folium_map):
|
||||
pedestrian_heat_view_data = get_view("pedestrianheat")
|
||||
heat_gdf = gpd.GeoDataFrame(pedestrian_heat_view_data, columns=['latitude', 'longitude', 'year'])
|
||||
add_heat_year_toggle(heat_gdf, folium_map, name="pedestrians")
|
||||
|
||||
|
||||
# Utilities ===========================================================================================================
|
||||
|
||||
def save_map_as_html(folium_map, name):
|
||||
folium_map.save(f"{name}.html")
|
||||
logger.info(f"Succesfully saved map {name}.")
|
||||
@ -232,6 +251,27 @@ def setup_views():
|
||||
create_pedestrian_heat_view()
|
||||
|
||||
|
||||
def generate_hue_gradient(hue, num_colors):
|
||||
if num_colors < 2:
|
||||
num_colors = 2
|
||||
gradient = {}
|
||||
for i in range(num_colors):
|
||||
lightness = 0.1 + 0.8 * (i / (num_colors - 1))
|
||||
saturation = 0.1 + 0.8 * (i / (num_colors - 1))
|
||||
rgb = colorsys.hls_to_rgb(hue, lightness, saturation)
|
||||
gradient[i / (num_colors - 1)] = '#{:02x}{:02x}{:02x}'.format(int(rgb[0]*255), int(rgb[1]*255), int(rgb[2]*255))
|
||||
return gradient
|
||||
|
||||
def generate_contrasting_gradient(num_colors):
|
||||
cmap = plt.get_cmap('viridis') # viridis is a map with contrasting colors
|
||||
gradient = {}
|
||||
for i in range(num_colors):
|
||||
rgba = cmap(i / (num_colors - 1))
|
||||
gradient[i / (num_colors - 1)] = '#{:02x}{:02x}{:02x}'.format(int(rgba[0]*255), int(rgba[1]*255), int(rgba[2]*255))
|
||||
return gradient
|
||||
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
time_map = folium.Map(
|
||||
location=zurich_coordinates,
|
||||
|
||||
Reference in New Issue
Block a user