feat(eda.correlation): move nan corr values to the bottom

sfu-db · Sep 4, 2020 · 4bba52e · 4bba52e
1 parent 64be889
commit 4bba52e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dataprep/eda/correlation/compute/univariate.py b/dataprep/eda/correlation/compute/univariate.py
@@ -134,6 +134,7 @@ def _corr_filter(
         # pylint: enable=invalid-unary-operand-type
 
     sorted_idx = np.argsort(corrs)
+    sorted_idx = np.roll(sorted_idx, np.count_nonzero(np.isnan(corrs)))
     sorted_corrs = corrs[sorted_idx]
 
     if value_range is not None: