adding cells mandatory for capacity analysis

queries/process_trx.py

@@ -63,6 +63,13 @@ TRX_BTS_COLUMNS = [
     "CBCs",
     "TotalChannels",
     "Signal",
+    "number_tch_per_cell",
+    "number_sd_per_cell",
+    "number_bcch_per_cell",
+    "number_ccch_per_cell",
+    "number_cbc_per_cell",
+    "number_total_channels_per_cell",
+    "number_signals_per_cell",
 ]
 
 
@@ -127,7 +134,9 @@ def process_trx_with_bts_name(file_path: str):
 
     df_bts = process_small_bts_data(file_path=file_path)
 
-    df_trx_bts_name = pd.merge(df_gsm_trx, df_bts, on="ID_BTS", how="left")
+    df_trx_bts_name: pd.DataFrame = pd.merge(
+        df_gsm_trx, df_bts, on="ID_BTS", how="left"
+    )
 
     # Filter columns strictly by names like "channelXType"
     channel_columns = [
@@ -173,6 +182,27 @@ def process_trx_with_bts_name(file_path: str):
     df_trx_bts_name["Signal"] = (
         df_trx_bts_name["BCCHs"] + df_trx_bts_name["CCCHs"] + df_trx_bts_name["CBCs"]
     )
+    df_trx_bts_name["number_tch_per_cell"] = df_trx_bts_name.groupby("ID_BTS")[
+        "TCHs"
+    ].transform("sum")
+    df_trx_bts_name["number_sd_per_cell"] = df_trx_bts_name.groupby("ID_BTS")[
+        "SDs"
+    ].transform("sum")
+    df_trx_bts_name["number_bcch_per_cell"] = df_trx_bts_name.groupby("ID_BTS")[
+        "BCCHs"
+    ].transform("sum")
+    df_trx_bts_name["number_ccch_per_cell"] = df_trx_bts_name.groupby("ID_BTS")[
+        "CCCHs"
+    ].transform("sum")
+    df_trx_bts_name["number_cbc_per_cell"] = df_trx_bts_name.groupby("ID_BTS")[
+        "CBCs"
+    ].transform("sum")
+    df_trx_bts_name["number_total_channels_per_cell"] = df_trx_bts_name.groupby(
+        "ID_BTS"
+    )["TotalChannels"].transform("sum")
+    df_trx_bts_name["number_signals_per_cell"] = df_trx_bts_name.groupby("ID_BTS")[
+        "Signal"
+    ].transform("sum")
 
     df_trx_bts_name = df_trx_bts_name[TRX_BTS_COLUMNS]