From 87d9a631d6403645f93f8fc9b68dc75b727d217d Mon Sep 17 00:00:00 2001 From: Timo Wentong Lin Date: Wed, 28 Aug 2024 12:18:28 +0200 Subject: [PATCH] fix: avoid warning message in corr metadata when metadata is correlated the case might happen that subgroups have 1 or less members, this results in a warning --- ggmap/correlations.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/ggmap/correlations.py b/ggmap/correlations.py index 43ad664..b1bf0dd 100644 --- a/ggmap/correlations.py +++ b/ggmap/correlations.py @@ -188,7 +188,6 @@ def correlate_metadata(metadata, meta, all_columns = _clear_metadata( metadata, categorials, ordinals, intervals, dates, err, for_metadata_correlation=True) - summary = dict() # start computing correlations # if err is not None: @@ -237,6 +236,11 @@ def correlate_metadata(metadata, groups = [g.dropna().values for n, g in meta.groupby(column_a)[column_b]] + + if any(len(group) <= 1 for group in groups): + print(f"Info: skipping correlation due to insufficient data: {column_a} vs {column_b}") + continue + f_, p_ = f_oneway(*groups) df_n = len(np.hstack(groups)) - len(groups)