1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- from typing import (
- Any,
- Callable,
- Iterable,
- Literal,
- overload,
- )
- import numpy as np
- from pandas._typing import npt
- _BinOp = Callable[[Any, Any], Any]
- _BoolOp = 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 = ...,
- false_values: Iterable = ...,
- 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]: ...
|