Use only numerical columns for correlation matrix (#45)

* Use only numerical column for correlation matrix * Use pandas isnumeric function instead of checking for np.number
equinor · Aug 7, 2019 · 6da7651 · 6da7651
1 parent 9416c5b
commit 6da7651
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/webviz_subsurface/containers/_parameter_distribution.py b/webviz_subsurface/containers/_parameter_distribution.py
@@ -319,7 +319,8 @@ def render_scatter(ens1, x_col, ens2, y_col, color, density):
 @cache.memoize(timeout=cache.TIMEOUT)
 def render_matrix(ensemble_path):
 
-    data = get_parameters(ensemble_path)
+    data = get_parameters(ensemble_path).apply(pd.to_numeric, errors='coerce')\
+                                        .dropna(how='all', axis='columns')
     values = list(data.corr().values)
 
     data = {