fix greenspace and add more schools

2026-03-25 08:06:16 +00:00 · 2026-03-25 08:06:16 +00:00 · 9cd2b8849c
commit 9cd2b8849c
parent 3a3e249bdd
4 changed files with 161 additions and 61 deletions
--- a/pipeline/transform/school_proximity.py
+++ b/pipeline/transform/school_proximity.py
@ -60,9 +60,14 @@ def main():
    # Load all postcodes for proximity counting
    postcodes = arcgis.rename({"lng": "lon"})

-    result = count_pois_per_postcode(
+    counts_5km = count_pois_per_postcode(
        postcodes, schools, radius_km=5, groups=SCHOOL_GROUPS
    )
+    counts_2km = count_pois_per_postcode(
+        postcodes, schools, radius_km=2, groups=SCHOOL_GROUPS
+    )
+
+    result = counts_5km.join(counts_2km, on="postcode")

    result.write_parquet(args.output)
    size_mb = args.output.stat().st_size / (1024 * 1024)