quantile add

Author: attack68

pandas/io/formats/style.py

@@ -1734,6 +1734,7 @@ def highlight_range(
         --------
         Styler.highlight_max:
         Styler.highlight_min:
+        Styler.highlight_quantile:
 
         Notes
         -----
@@ -1952,25 +1953,15 @@ def _highlight_func(
             return np.where(pd.isna(data).values, props, "")
         elif highlight == "range":
             l, h = kwargs.get("start"), kwargs.get("stop")
-            in_l = data >= l if l is not None else np.full_like(data, True, dtype=bool)
-            in_h = data <= h if h is not None else np.full_like(data, True, dtype=bool)
-            return np.where(in_l & in_h, props, "")
+            ge_l = data >= l if l is not None else np.full_like(data, True, dtype=bool)
+            le_h = data <= h if h is not None else np.full_like(data, True, dtype=bool)
+            return np.where(ge_l & le_h, props, "")
         elif highlight == "quantile":
             l, h = kwargs.get("q_low"), kwargs.get("q_high")
             if l > 0:
                 q, tgt_label = [0, l, h], 1
             else:
                 q, tgt_label = [0, h], 0
-            # if l > 0 and h < 1:
-            #     q, tgt_label = [0, l, h, 1], 1
-            # elif l > 0 and h == 1:
-            #     q, tgt_label = [0, l, h], 1
-            # elif l == 0 and h < 1:
-            #     q, tgt_label = [0, h, 1], 0
-            # elif l == 0 and h == 1:
-            #     q, tgt_label = [0, 1], 0
-            # else:
-            #     raise ValueError("'q_low' and 'q_high' must be in range [0,1]")
             if kwargs.get("axis_") is None:
                 shape = data.values.shape
                 labels = pd.qcut(data.values.ravel(), q=q, labels=False).reshape(shape)