fix bug when few different values (#52)

societe-generale · May 27, 2020 · 9496bbc · 9496bbc
1 parent 50a82c6
commit 9496bbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aikit/plot.py b/aikit/plot.py
@@ -152,7 +152,7 @@ def conditional_boxplot(df, var, explaining_var, nb_quantiles=10, use_rank=True,
         var_bis = var
         explaining_var_bis = explaining_var
 
-    df_copy["_quantile"] = pd.qcut(df_copy[explaining_var_bis], q=nb_quantiles)
+    df_copy["_quantile"] = pd.qcut(df_copy[explaining_var_bis], q=nb_quantiles, duplicates="drop")
 
     positions = [np.median(sub_df[explaining_var_bis].values) for _, sub_df in df_copy.groupby("_quantile")]
     if use_rank: