From 396240640d5d5d9b3c0e8c116eacbdfdd41c51f1 Mon Sep 17 00:00:00 2001 From: Miki Watanabe Date: Thu, 23 Mar 2023 06:01:03 -0400 Subject: [PATCH] BUG: quantile only working with categorical NAs --- pandas/core/array_algos/quantile.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pandas/core/array_algos/quantile.py b/pandas/core/array_algos/quantile.py index 7647f321e1d4b..831b0b410e20c 100644 --- a/pandas/core/array_algos/quantile.py +++ b/pandas/core/array_algos/quantile.py @@ -217,7 +217,7 @@ def _nanpercentile( result = result.astype(values.dtype, copy=False) return result else: - return np.percentile( + result = np.percentile( values, qs, axis=1, @@ -226,3 +226,4 @@ def _nanpercentile( # "int", "Dict[str, str]" [call-overload] **{np_percentile_argname: interpolation}, # type: ignore[call-overload] ) + return result.astype(values.dtype, copy=False)