Update functions.py

functions.py

@@ -93,6 +93,16 @@ def encoder_range(temps):
     redf = np.logical_not(temps<=boundary_list) 
     hift = np.concatenate((np.roll(redf, -1)[:, :-1], np.full((temps.shape[0], 1), False)), axis = 1)
     cat = np.nonzero(np.not_equal(redf,hift))
+    air_pollution_level = ['Good', 'Moderate', 'Unhealthy for sensitive Groups','Unhealthy' ,'Very Unhealthy', 'Hazardous']
+    level = [air_pollution_level[el] for el in cat[1]]
+    return level
+
+def get_aplevel(temps:np.ndarray) -> list:
+    boundary_list = np.array([0, 50, 100, 150, 200, 300]) # assert temps.shape == [x, 1]
+    redf = np.logical_not(temps<=boundary_list) # temps.shape[0] x boundary_list.shape[0] ndarray
+    hift = np.concatenate((np.roll(redf, -1)[:, :-1], np.full((temps.shape[0], 1), False)), axis = 1)
+    cat = np.nonzero(np.not_equal(redf,hift))
+
     air_pollution_level = ['Good', 'Moderate', 'Unhealthy for sensitive Groups','Unhealthy' ,'Very Unhealthy', 'Hazardous']
     level = [air_pollution_level[el] for el in cat[1]]
     return level