seb/group-1-db23
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small refactor of map.py.

Browse Source
This commit is contained in:
Sebastian Lenzlinger 2024-01-05 15:45:39 +01:00
parent 0f73f829aa
commit 617cd2d1f4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
analysis/map.py
View File

@ -137,7 +137,7 @@ def create_heat_map_with_time():
heat_df = gpd.GeoDataFrame(heat_view_data, columns=['latitude', 'longitude', 'year']) heat_df = gpd.GeoDataFrame(heat_view_data, columns=['latitude', 'longitude', 'year'])
assert not heat_df.empty, f" Heat Dataframe is empty: {heat_df.head(5)}" assert not heat_df.empty, f" Heat Dataframe is empty: {heat_df.head(5)}"
add_heat_map_time(heat_df) add_heat_map_time(heat_df, interactive_map)
#interactive_map.save("test.html") #interactive_map.save("test.html")
add_signaled_speeds(interactive_map) add_signaled_speeds(interactive_map)
@ -145,7 +145,7 @@ def create_heat_map_with_time():
folium.LayerControl(collapsed=True).add_to(interactive_map) folium.LayerControl(collapsed=True).add_to(interactive_map)
def add_heat_map_time(heat_df): def add_heat_map_time(heat_df, 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.1] for index, row in heat_df[heat_df['year'] == i].iterrows()] for
i in range(2011, 2023)] i in range(2011, 2023)]
index = [2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022] index = [2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022]
@ -161,7 +161,7 @@ def add_heat_map_time(heat_df):
blur=1, blur=1,
index=index, index=index,
name="Accident Heatmap") name="Accident Heatmap")
hm.add_to(interactive_map) hm.add_to(map)
def add_signaled_speeds(map): def add_signaled_speeds(map):