Merge branch 'main' into propagate_fovs

angelolab · Feb 9, 2023 · 0b37d7b · 0b37d7b
2 parents 5b7794b + 8d698a0
commit 0b37d7b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ark/phenotyping/cluster_helpers.py b/ark/phenotyping/cluster_helpers.py
@@ -279,7 +279,12 @@ def normalize_data(self):
         """
         # only 99.9% normalize on the columns provided
         cell_data_sub = self.cell_data[self.columns].copy()
-        cell_data_sub = cell_data_sub.div(cell_data_sub.quantile(0.999))
+
+        # compute the 99.9% normalization values, ignoring zeros
+        cell_norm_vals = cell_data_sub.replace(0, np.nan).quantile(q=0.999, axis=0)
+
+        # divide cell_data_sub by normalization values
+        cell_data_sub = cell_data_sub.div(cell_norm_vals)
 
         # assign back to cell_data
         self.cell_data[self.columns] = cell_data_sub