Adding Distance to all neighbors

queries/process_neighbors.py

@@ -2,6 +2,7 @@ import pandas as pd
 from geopy.distance import geodesic
 
 from queries.process_gsm import process_gsm_data
+from queries.process_wcdma import process_wcdma_data
 from utils.convert_to_excel import convert_dfs, save_dataframe
 from utils.utils_vars import UtilsVars
 
@@ -31,11 +32,15 @@ BTS_TARGET = [
 WCEL_SOURCE = [
     "ID_WCEL",
     "name",
+    "Longitude",
+    "Latitude",
 ]
 
 WCEL_TARGET = [
     "lac_id",
     "name",
+    "Longitude",
+    "Latitude",
 ]
 
 
@@ -132,7 +137,7 @@ def process_neighbors_data(file_path: str):
     df_adce_final = check_symmetry(df_adce_final)
 
     # create distance column
-    df_adce_final["distance"] = df_adce_final.apply(
+    df_adce_final["distance_km"] = df_adce_final.apply(
         lambda row: geodesic(
             (row["SOURCE_Latitude"], row["SOURCE_Longitude"]),
             (row["TARGET_Latitude"], row["TARGET_Longitude"]),
@@ -153,13 +158,13 @@ def process_neighbors_data(file_path: str):
     df_adjs = df_adjs[ADJS_INITIAL_COLUMNS]
 
     # process WCEL DATA
-    df_wcel = dfs["WCEL"]
-    df_wcel.columns = df_wcel.columns.str.replace(r"[ ]", "", regex=True)
+    df_wcel = process_wcdma_data(file_path)
+
     df_wcel["ID_WCEL"] = (
         df_wcel[["RNC", "WBTS", "WCEL"]].astype(str).apply("_".join, axis=1)
     )
     df_wcel["lac_id"] = df_wcel[["LAC", "CId"]].astype(str).apply("_".join, axis=1)
-    df_wcel = df_wcel[["ID_WCEL", "lac_id", "name"]]
+    df_wcel = df_wcel[["ID_WCEL", "lac_id", "name", "Longitude", "Latitude"]]
 
     df_wcel_source = df_wcel[WCEL_SOURCE]
     df_wcel_source.rename(columns={"name": "SOURCE_NAME"}, inplace=True)
@@ -176,6 +181,14 @@ def process_neighbors_data(file_path: str):
         columns={"ID_WCEL": "SOURCE_ID", "lac_id": "TARGET_LAC_ID"}, inplace=True
     )
     df_adjs_final = check_symmetry(df_adjs_final)
+    # create distance column
+    df_adjs_final["distance_km"] = df_adjs_final.apply(
+        lambda row: geodesic(
+            (row["Latitude_x"], row["Longitude_x"]),
+            (row["Latitude_y"], row["Longitude_y"]),
+        ).kilometers,
+        axis=1,
+    )
 
     # process ADJI DATA
     df_adji = dfs["ADJI"]
@@ -197,6 +210,14 @@ def process_neighbors_data(file_path: str):
         columns={"ID_WCEL": "SOURCE_ID", "lac_id": "TARGET_LAC_ID"}, inplace=True
     )
     df_adji_final = check_symmetry(df_adji_final)
+    # create distance column
+    df_adji_final["distance_km"] = df_adji_final.apply(
+        lambda row: geodesic(
+            (row["Latitude_x"], row["Longitude_x"]),
+            (row["Latitude_y"], row["Longitude_y"]),
+        ).kilometers,
+        axis=1,
+    )
 
     # process ADJG DATA
     df_adjg = dfs["ADJG"]
@@ -217,6 +238,14 @@ def process_neighbors_data(file_path: str):
     df_adjg_final.rename(
         columns={"ID_WCEL": "SOURCE_ID", "lac_id": "TARGET_LAC_ID"}, inplace=True
     )
+    # create distance column
+    df_adjg_final["distance_km"] = df_adjg_final.apply(
+        lambda row: geodesic(
+            (row["Latitude_x"], row["Longitude_x"]),
+            (row["Latitude_y"], row["Longitude_y"]),
+        ).kilometers,
+        axis=1,
+    )
 
     # process ADJW DATA
     df_adjw = dfs["ADJW"]
@@ -235,6 +264,15 @@ def process_neighbors_data(file_path: str):
     df_adjw_final.rename(
         columns={"ID_BTS": "SOURCE_ID", "lac_id": "TARGET_LAC_ID"}, inplace=True
     )
+    df_adjw_final = check_symmetry(df_adjw_final)
+    # create distance column
+    df_adjw_final["distance_km"] = df_adjw_final.apply(
+        lambda row: geodesic(
+            (row["Latitude_x"], row["Longitude_x"]),
+            (row["Latitude_y"], row["Longitude_y"]),
+        ).kilometers,
+        axis=1,
+    )
 
     # save_dataframe(df_adjw_final, "ADJW")