Call city names function

Special-Unitary-Group · Aug 26, 2024 · 1e236b2 · 1e236b2
1 parent 6a7b5ba
commit 1e236b2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/cities.py b/cities.py
@@ -6,6 +6,14 @@ class Cities:
     def __init__(self, grouped_df):
         self.grouped_df = grouped_df
 
+    def get_city_data(self, city_name):
+        if city_name not in self.grouped_df["City"].unique():
+            raise ValueError(f"City '{city_name}' not found in the dataset.")
+        city_df = self.grouped_df[self.grouped_df["City"] == city_name]
+        city_df = city_df.drop(["City"], axis=1)
+        return city_df
+
+
     def apodaca(self):
         df_apodaca = self.grouped_df[self.grouped_df["City"] == "Apodaca"]
         df_apodaca = df_apodaca.drop(["City"], axis=1)