Transform additionnal plots

Author: guillaume-vignal

shapash/plots/plot_univariate.py

@@ -165,9 +165,11 @@ def plot_continuous_distribution(
     else:
         style_dict = define_style(get_palette(palette_name))
 
-    lower_quantile = df_all[col].quantile(0.005)
-    upper_quantile = df_all[col].quantile(0.995)
-    filtered_data = df_all[(df_all[col] > lower_quantile) & (df_all[col] < upper_quantile)].copy()
+    filtered_data = df_all.copy()
+    if len(filtered_data) > 200:
+        lower_quantile = filtered_data[col].quantile(0.005)
+        upper_quantile = filtered_data[col].quantile(0.995)
+        filtered_data = filtered_data[(filtered_data[col] > lower_quantile) & (filtered_data[col] < upper_quantile)]
 
     # Initialize the figure
     fig = go.Figure()

tests/unit_tests/report/test_plots.py

@@ -4,7 +4,6 @@
 import numpy as np
 import pandas as pd
 
-from shapash.report.common import VarType
 from shapash.plots.plot_univariate import (
     plot_distribution,
     plot_categorical_distribution,
@@ -56,7 +55,7 @@ def test_plot_distribution_3(self, mock_plot_cat, mock_plot_cont):
     def test_plot_continuous_distribution_1(self):
         df = pd.DataFrame(
             {
-                "int_data": [10, 20, 30, 40],
+                "int_data": [10, 20, 30, 40, 50],
             }
         )
         fig = plot_continuous_distribution(df, "int_data")
@@ -67,8 +66,8 @@ def test_plot_continuous_distribution_1(self):
     def test_plot_continuous_distribution_2(self):
         df = pd.DataFrame(
             {
-                "int_data": [10, 20, 30, 40, 50, 30, 20, 0],
-                "data_train_test": ["train", "train", "train", "train", "test", "test", "test", "test"],
+                "int_data": [10, 20, 30, 40, 50, 30, 20, 0, 10, 20],
+                "data_train_test": ["train", "train", "train", "train", "train", "test", "test", "test", "test", "test"],
             }
         )
         fig = plot_continuous_distribution(df, "int_data", "data_train_test")

tests/unit_tests/utils/test_utils.py

@@ -67,7 +67,7 @@ def test_compute_digit_number_2(self):
 
     def test_compute_digit_number_3(self):
         t = compute_digit_number(0.000044)
-        assert t == 7
+        assert t == 8
 
     def test_truncate_str_1(self):
         t = truncate_str(12)