Minor cleanup

burggraaff · Aug 22, 2024 · fb7c960 · fb7c960
1 parent b1dab91
commit fb7c960
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/fpcup/geo.py b/fpcup/geo.py
@@ -9,6 +9,8 @@
 gpd.options.io_engine = "pyogrio"
 import h3pandas
 import numpy as np
+import pandas as pd
+pd.options.mode.chained_assignment = None
 from pandas import DataFrame, Series
 from shapely import Geometry, Point, Polygon
 from tqdm import tqdm

diff --git a/process_brp.py b/process_brp.py
@@ -26,7 +26,7 @@
 # Remove unnecessary columns and translate to English
 brp.drop(columns=["jaar", "status"], inplace=True)
 brp.rename({"gewas": "crop", "gewascode": "crop_code"}, axis=1, inplace=True)
-brp["category"].replace(fpcup.site.brp_categories_NL2EN, inplace=True)
+brp.replace({"category": fpcup.site.brp_categories_NL2EN}, inplace=True)
 
 # Add province information
 fpcup.geo.add_provinces(brp)
@@ -54,7 +54,7 @@
 
 # Select fpcup crops
 brp_fpcup = brp_agro.loc[brp_agro["crop"].isin(fpcup.crop.brp_crops_NL2EN)]
-brp_fpcup["crop"].replace(fpcup.crop.brp_crops_NL2EN, inplace=True)
+brp_fpcup.replace({"crop": fpcup.crop.brp_crops_NL2EN}, inplace=True)
 if args.verbose:
     print(f"Reduced file to crops listed in the FPCUP/BRP dictionary -- {len(brp_fpcup)} entries")