From 2ea599e262fa14a9f7383db327c5fba40691f9c2 Mon Sep 17 00:00:00 2001
From: Meng Li <34143965+Meng6@users.noreply.github.com>
Date: Thu, 5 Mar 2020 16:49:06 -0500
Subject: [PATCH] Replace NA with 0 for heart rate zone features

---
 src/features/fitbit_heartrate_metrics.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/features/fitbit_heartrate_metrics.py b/src/features/fitbit_heartrate_metrics.py
index f2921c8e..a5848106 100644
--- a/src/features/fitbit_heartrate_metrics.py
+++ b/src/features/fitbit_heartrate_metrics.py
@@ -43,6 +43,7 @@ if not heartrate_data.empty:
         for feature_name in list(set(["lengthoutofrange", "lengthfatburn", "lengthcardio", "lengthpeak"]) & set(metrics)):
             heartrate_zone = heartrate_data[heartrate_data["heartrate_zone"] == feature_name[6:]]
             heartrate_features["heartrate_" + day_segment + "_" + feature_name] = heartrate_zone.groupby(["local_date"])["device_id"].count() / num_rows_per_minute
+            heartrate_features.fillna(value={"heartrate_" + day_segment + "_" + feature_name: 0}, inplace=True)
 
         heartrate_features = heartrate_features.reset_index()