def quantile(
self,
*,
qs: Float64Index,
axis: int = 0,
transposed: bool = False,
interpolation="linear",
) -> ArrayManager:
# error: Value of type variable "ArrayLike" of "ensure_block_shape" cannot be
# "Union[ndarray, ExtensionArray]"
arrs = [ensure_block_shape(x, 2)
for x in self.arrays] # type: ignore[type-var]
assert axis == 1
# error: Value of type variable "ArrayLike" of "quantile_compat" cannot be
# "object"
new_arrs = [
quantile_compat(x, qs, interpolation,
axis=axis) # type: ignore[type-var]
for x in arrs
]
for i, arr in enumerate(new_arrs):
if arr.ndim == 2:
assert arr.shape[0] == 1, arr.shape
new_arrs[i] = arr[0]
axes = [qs, self._axes[1]]
return type(self)(new_arrs, axes)
def quantile(
self,
*,
qs: Float64Index,
axis: int = 0,
transposed: bool = False,
interpolation="linear",
) -> ArrayManager:
arrs = [ensure_block_shape(x, 2) for x in self.arrays]
assert axis == 1
new_arrs = [quantile_compat(x, qs, interpolation, axis=axis) for x in arrs]
for i, arr in enumerate(new_arrs):
if arr.ndim == 2:
assert arr.shape[0] == 1, arr.shape
new_arrs[i] = arr[0]
axes = [qs, self._axes[1]]
return type(self)(new_arrs, axes)
