magic

script.py

@@ -6,9 +6,10 @@ import numpy as np
 brands = None
 
 def process_data(df):
-    df["age"] = df.year.apply(lambda x: np.sqrt(2017-x))
-    df["sqrt_mileage"] = df.mileage.apply(lambda x: np.sqrt(x))
-    df["sqrt_engine_capacity"] = df.engine_capacity.apply(lambda x: np.sqrt(x))
+    df["age"] = 2018 - df["year"]
+    df["sqrt_age"] = df.age**0.7
+    df["sqrt_mileage"] = df.mileage ** 0.7
+    df["sqrt_engine_capacity"] = df.engine_capacity ** 0.7
     global brands
     if not brands:
         brands = df.brand.value_counts()[:35].index.tolist()