From 96394a046872802530433fe1d490155ab7e693a4 Mon Sep 17 00:00:00 2001
From: Meng Li <34143965+Meng6@users.noreply.github.com>
Date: Tue, 17 Dec 2019 17:07:35 -0500
Subject: [PATCH] Replace ZERO with NaN for heatmap

---
 src/visualization/compliance_heatmap.py | 3 +++
 src/visualization/heatmap_rows.py       | 5 ++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/visualization/compliance_heatmap.py b/src/visualization/compliance_heatmap.py
index 5fb96baf..1fa76f8a 100644
--- a/src/visualization/compliance_heatmap.py
+++ b/src/visualization/compliance_heatmap.py
@@ -37,5 +37,8 @@ else:
     phone_sensed_bins = phone_sensed_bins.resample("1D").asfreq().fillna(0)
     # get dates and compliance_matrix
     dates, compliance_matrix = getDatesComplianceMatrix(phone_sensed_bins)
+    # convert compliance_matrix from list to np.array and replace 0 with np.nan
+    compliance_matrix = np.asarray(compliance_matrix)
+    compliance_matrix = np.where(compliance_matrix == 0, np.nan, compliance_matrix)
     # get heatmap
     getComplianceHeatmap(dates, compliance_matrix, pid, snakemake.output[0], bin_size)
\ No newline at end of file
diff --git a/src/visualization/heatmap_rows.py b/src/visualization/heatmap_rows.py
index 4c794665..fa5d24aa 100644
--- a/src/visualization/heatmap_rows.py
+++ b/src/visualization/heatmap_rows.py
@@ -1,4 +1,5 @@
 import pandas as pd
+import numpy as np
 import plotly.io as pio
 import plotly.graph_objects as go
 import datetime
@@ -57,4 +58,6 @@ else:
 
     # Create heatmap
     hourly_row_count = getComplianceMatrix(dates, resampled_hourly_bins)
-    getHourlyRowCountHeatmap(dates, hourly_row_count, sensor_name, pid, snakemake.output[0])
+    hourly_row_count = np.asarray(hourly_row_count)
+    hourly_row_count = np.where(hourly_row_count == 0, np.nan, hourly_row_count)
+    getHourlyRowCountHeatmap(dates, hourly_row_count, sensor_name, pid, snakemake.output[0])
\ No newline at end of file