52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
|
from typing import (
|
||
|
|
Any,
|
||
|
|
Callable,
|
||
|
|
Iterable,
|
||
|
|
Literal,
|
||
|
|
TypeAlias,
|
||
|
|
overload,
|
||
|
|
)
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
from pandas._typing import npt
|
||
|
|
|
||
|
|
_BinOp: TypeAlias = Callable[[Any, Any], Any]
|
||
|
|
_BoolOp: TypeAlias = Callable[[Any, Any], bool]
|
||
|
|
|
||
|
|
def scalar_compare(
|
||
|
|
values: np.ndarray, # object[:]
|
||
|
|
val: object,
|
||
|
|
op: _BoolOp, # {operator.eq, operator.ne, ...}
|
||
|
|
) -> npt.NDArray[np.bool_]: ...
|
||
|
|
def vec_compare(
|
||
|
|
left: npt.NDArray[np.object_],
|
||
|
|
right: npt.NDArray[np.object_],
|
||
|
|
op: _BoolOp, # {operator.eq, operator.ne, ...}
|
||
|
|
) -> npt.NDArray[np.bool_]: ...
|
||
|
|
def scalar_binop(
|
||
|
|
values: np.ndarray, # object[:]
|
||
|
|
val: object,
|
||
|
|
op: _BinOp, # binary operator
|
||
|
|
) -> np.ndarray: ...
|
||
|
|
def vec_binop(
|
||
|
|
left: np.ndarray, # object[:]
|
||
|
|
right: np.ndarray, # object[:]
|
||
|
|
op: _BinOp, # binary operator
|
||
|
|
) -> np.ndarray: ...
|
||
|
|
@overload
|
||
|
|
def maybe_convert_bool(
|
||
|
|
arr: npt.NDArray[np.object_],
|
||
|
|
true_values: Iterable | None = None,
|
||
|
|
false_values: Iterable | None = None,
|
||
|
|
convert_to_masked_nullable: Literal[False] = ...,
|
||
|
|
) -> tuple[np.ndarray, None]: ...
|
||
|
|
@overload
|
||
|
|
def maybe_convert_bool(
|
||
|
|
arr: npt.NDArray[np.object_],
|
||
|
|
true_values: Iterable = ...,
|
||
|
|
false_values: Iterable = ...,
|
||
|
|
*,
|
||
|
|
convert_to_masked_nullable: Literal[True],
|
||
|
|
) -> tuple[np.ndarray, np.ndarray]: ...
|