Change column names (#53)

opengeos · Jun 6, 2024 · 552f816 · 552f816
1 parent bad5670
commit 552f816
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 18 deletions.
diff --git a/lidar/filling.py b/lidar/filling.py
@@ -157,35 +157,35 @@ def write_dep_csv(dep_list, csv_file):
     """
     csv = open(csv_file, "w")
     header = (
-        "region-id"
+        "region_id"
         + ","
         + "count"
         + ","
         + "area"
         + ","
         + "volume"
         + ","
-        + "avg-depth"
+        + "avg_depth"
         + ","
-        + "max-depth"
+        + "max_depth"
         + ","
-        + "min-elev"
+        + "min_elev"
         + ","
-        + "max-elev"
+        + "max_elev"
         + ","
         + "perimeter"
         + ","
-        + "major-axis"
+        + "major_axis"
         + ","
-        + "minor-axis"
+        + "minor_axis"
         + ","
         + "elongatedness"
         + ","
         + "eccentricity"
         + ","
         + "orientation"
         + ","
-        + "area-bbox-ratio"
+        + "area_bbox_ratio"
     )
 
     csv.write(header + "\n")
@@ -376,7 +376,7 @@ def ExtractSinks(
     print("Converting raster to vector ...")
     polygonize(out_region, out_vec_file)
 
-    gdf = join_csv_to_gdf(out_vec_file, out_csv_file, "id", "region-id")
+    gdf = join_csv_to_gdf(out_vec_file, out_csv_file, "id", "region_id")
     gdf.drop(columns=["id"], inplace=True)
     gdf.to_file(out_gpkg, driver="GPKG")
 

diff --git a/lidar/slicing.py b/lidar/slicing.py
@@ -555,31 +555,31 @@ def write_dep_csv(dep_list, csv_file):
         + ","
         + "volume"
         + ","
-        + "avg-depth"
+        + "avg_depth"
         + ","
-        + "max-depth"
+        + "max_depth"
         + ","
-        + "min-elev"
+        + "min_elev"
         + ","
-        + "max-elev"
+        + "max_elev"
         + ","
-        + "children-id"
+        + "children_id"
         + ","
-        + "region-id"
+        + "region_id"
         + ","
         + "perimeter"
         + ","
-        + "major-axis"
+        + "major_axis"
         + ","
-        + "minor-axis"
+        + "minor_axis"
         + ","
         + "elongatedness"
         + ","
         + "eccentricity"
         + ","
         + "orientation"
         + ","
-        + "area-bbox-ratio"
+        + "area_bbox_ratio"
     )
     csv.write(header + "\n")
     for dep in dep_list: