Skip to content
Snippets Groups Projects
Verified Commit 8f5d094a authored by Marco Aceti's avatar Marco Aceti
Browse files

Drop cancelled stops and trains in trajectories map

parent 8d6909fd
No related branches found
No related tags found
No related merge requests found
Pipeline #2829 passed
......@@ -274,6 +274,9 @@ def build_map(st: pd.DataFrame, df: pd.DataFrame) -> None:
"""
m = folium.Map(**MAP_KWARGS)
# Drop cancelled stops and trains
df = df.loc[(df.stop_type != "C") & (df.cancelled == False)].copy()
logging.info("Generating GeoJSON features...")
features = Parallel(n_jobs=-1, verbose=5)(
train_stop_geojson(st, train_df) for _, train_df in df.groupby("train_hash")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment