betterdatascience · December 11, 2020 08:39 · Dec 11, 2020
diff --git a/maps.py b/maps.py
@@ -0,0 +1,35 @@
+def generate_color(magnitude):
+    if magnitude <= 5:
+        c_outline, c_fill = '#ffda79', '#ffda79'
+        m_opacity, f_opacity = 0.2, 0.1
+    else:
+        c_outline, c_fill = '#c0392b', '#e74c3c'
+        m_opacity, f_opacity = 1, 1
+    return c_outline, c_fill, m_opacity, f_opacity
+
+def generate_popup(magnitude, depth):
+    return f'''<strong>Magnitude:</strong> {magnitude}<br><strong>Depth:</strong> {depth} km'''
+
+
+quake_map = folium.Map(
+    location=[-16.495477, 174.9663341],
+    zoom_start=5,
+    tiles='Stamen Terrain',
+    width=1024,
+    height=600
+)
+
+for _, row in df.iterrows():
+    c_outline, c_fill, m_opacity, f_opacity = generate_color(row['mag'])
+    folium.CircleMarker(
+        location=[row['lat'], row['long']],
+        popup=generate_popup(row['mag'], row['depth']),
+        color=c_outline,
+        fill=True,
+        fillColor=c_fill,
+        opacity=m_opacity,
+        fillOpacity=f_opacity,
+        radius=(row['mag'] ** 2) / 3
+    ).add_to(quake_map)
+
+quake_map