| 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: ... | |
| 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]: ... | |
| 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]: ... | |