fix radius filter

alexanderquispe · Jul 8, 2023 · 806351a · 806351a
1 parent 42da25e
commit 806351a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,7 +2,7 @@ data/
 *__pycache__
 examples/continent
 examples/country
-examples/subregion
+examples/sub_region_pbf
 examples/data-pbf
 
 *algo*
diff --git a/osrmareas/areas.py b/osrmareas/areas.py
@@ -77,8 +77,7 @@ def get_routes(
 			pd.concat([all_routes_df, get_osrm_route(center1, final1)])
 
 	if filter_km:
-		rows_inside = all_routes_df['dist_driving_km']  < radius_km
-		all_routes_df_crop = all_routes_df.loc[rows_inside]
+		all_routes_df_crop = all_routes_df.query('dist_driving_km <= @radius_km')
 		return all_routes_df_crop
 
 	return all_routes_df