Update functions.py

functions.py

@@ -3,7 +3,7 @@ import requests
 import os
 import joblib
 import pandas as pd
-
+import numpy as np
 import json
 
 from dotenv import load_dotenv
@@ -86,4 +86,13 @@ def get_weather_df(data):
 def timestamp_2_time(x):
     dt_obj = datetime.strptime(str(x), '%Y-%m-%d')
     dt_obj = dt_obj.timestamp() * 1000
-    return int(dt_obj)
+    return int(dt_obj)
+
+def encoder_range(temps):
+    boundary_list = np.array([0, 50, 100, 150, 200, 300]) 
+    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