from collections.abc import Sequence
from typing import Any, overload, TypeVar, Literal, SupportsIndex
import numpy as np
from numpy import (
number,
uint64,
int_,
int64,
intp,
float16,
floating,
complexfloating,
object_,
generic,
_OrderKACF,
_OrderACF,
_ModeKind,
_PartitionKind,
_SortKind,
_SortSide,
_CastingKind,
)
from numpy._typing import (
DTypeLike,
_DTypeLike,
ArrayLike,
_ArrayLike,
NDArray,
_ShapeLike,
_Shape,
_ArrayLikeBool_co,
_ArrayLikeUInt_co,
_ArrayLikeInt_co,
_ArrayLikeFloat_co,
_ArrayLikeComplex_co,
_ArrayLikeObject_co,
_IntLike_co,
_BoolLike_co,
_ComplexLike_co,
_NumberLike_co,
_ScalarLike_co,
)
_SCT = TypeVar("_SCT", bound=generic)
_SCT_uifcO = TypeVar("_SCT_uifcO", bound=number[Any] | object_)
_ArrayType = TypeVar("_ArrayType", bound=NDArray[Any])
__all__: list[str]
@overload
def take(
a: _ArrayLike[_SCT],
indices: _IntLike_co,
axis: None = ...,
out: None = ...,
mode: _ModeKind = ...,
) -> _SCT: ...
@overload
def take(
a: ArrayLike,
indices: _IntLike_co,
axis: None | SupportsIndex = ...,
out: None = ...,
mode: _ModeKind = ...,
) -> Any: ...
@overload
def take(
a: _ArrayLike[_SCT],
indices: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
out: None = ...,
mode: _ModeKind = ...,
) -> NDArray[_SCT]: ...
@overload
def take(
a: ArrayLike,
indices: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
out: None = ...,
mode: _ModeKind = ...,
) -> NDArray[Any]: ...
@overload
def take(
a: ArrayLike,
indices: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
out: _ArrayType = ...,
mode: _ModeKind = ...,
) -> _ArrayType: ...
@overload
def reshape(
a: _ArrayLike[_SCT],
/,
shape: _ShapeLike = ...,
order: _OrderACF = ...,
*,
newshape: _ShapeLike = ...,
copy: None | bool = ...,
) -> NDArray[_SCT]: ...
@overload
def reshape(
a: ArrayLike,
/,
shape: _ShapeLike = ...,
order: _OrderACF = ...,
*,
newshape: _ShapeLike = ...,
copy: None | bool = ...,
) -> NDArray[Any]: ...
@overload
def choose(
a: _IntLike_co,
choices: ArrayLike,
out: None = ...,
mode: _ModeKind = ...,
) -> Any: ...
@overload
def choose(
a: _ArrayLikeInt_co,
choices: _ArrayLike[_SCT],
out: None = ...,
mode: _ModeKind = ...,
) -> NDArray[_SCT]: ...
@overload
def choose(
a: _ArrayLikeInt_co,
choices: ArrayLike,
out: None = ...,
mode: _ModeKind = ...,
) -> NDArray[Any]: ...
@overload
def choose(
a: _ArrayLikeInt_co,
choices: ArrayLike,
out: _ArrayType = ...,
mode: _ModeKind = ...,
) -> _ArrayType: ...
@overload
def repeat(
a: _ArrayLike[_SCT],
repeats: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
) -> NDArray[_SCT]: ...
@overload
def repeat(
a: ArrayLike,
repeats: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
) -> NDArray[Any]: ...
def put(
a: NDArray[Any],
ind: _ArrayLikeInt_co,
v: ArrayLike,
mode: _ModeKind = ...,
) -> None: ...
@overload
def swapaxes(
a: _ArrayLike[_SCT],
axis1: SupportsIndex,
axis2: SupportsIndex,
) -> NDArray[_SCT]: ...
@overload
def swapaxes(
a: ArrayLike,
axis1: SupportsIndex,
axis2: SupportsIndex,
) -> NDArray[Any]: ...
@overload
def transpose(
a: _ArrayLike[_SCT],
axes: None | _ShapeLike = ...
) -> NDArray[_SCT]: ...
@overload
def transpose(
a: ArrayLike,
axes: None | _ShapeLike = ...
) -> NDArray[Any]: ...
@overload
def matrix_transpose(x: _ArrayLike[_SCT]) -> NDArray[_SCT]: ...
@overload
def matrix_transpose(x: ArrayLike) -> NDArray[Any]: ...
@overload
def partition(
a: _ArrayLike[_SCT],
kth: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
kind: _PartitionKind = ...,
order: None | str | Sequence[str] = ...,
) -> NDArray[_SCT]: ...
@overload
def partition(
a: ArrayLike,
kth: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
kind: _PartitionKind = ...,
order: None | str | Sequence[str] = ...,
) -> NDArray[Any]: ...
def argpartition(
a: ArrayLike,
kth: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
kind: _PartitionKind = ...,
order: None | str | Sequence[str] = ...,
) -> NDArray[intp]: ...
@overload
def sort(
a: _ArrayLike[_SCT],
axis: None | SupportsIndex = ...,
kind: None | _SortKind = ...,
order: None | str | Sequence[str] = ...,
*,
stable: None | bool = ...,
) -> NDArray[_SCT]: ...
@overload
def sort(
a: ArrayLike,
axis: None | SupportsIndex = ...,
kind: None | _SortKind = ...,
order: None | str | Sequence[str] = ...,
*,
stable: None | bool = ...,
) -> NDArray[Any]: ...
def argsort(
a: ArrayLike,
axis: None | SupportsIndex = ...,
kind: None | _SortKind = ...,
order: None | str | Sequence[str] = ...,
*,
stable: None | bool = ...,
) -> NDArray[intp]: ...
@overload
def argmax(
a: ArrayLike,
axis: None = ...,
out: None = ...,
*,
keepdims: Literal[False] = ...,
) -> intp: ...
@overload
def argmax(
a: ArrayLike,
axis: None | SupportsIndex = ...,
out: None = ...,
*,
keepdims: bool = ...,
) -> Any: ...
@overload
def argmax(
a: ArrayLike,
axis: None | SupportsIndex = ...,
out: _ArrayType = ...,
*,
keepdims: bool = ...,
) -> _ArrayType: ...
@overload
def argmin(
a: ArrayLike,
axis: None = ...,
out: None = ...,
*,
keepdims: Literal[False] = ...,
) -> intp: ...
@overload
def argmin(
a: ArrayLike,
axis: None | SupportsIndex = ...,
out: None = ...,
*,
keepdims: bool = ...,
) -> Any: ...
@overload
def argmin(
a: ArrayLike,
axis: None | SupportsIndex = ...,
out: _ArrayType = ...,
*,
keepdims: bool = ...,
) -> _ArrayType: ...
@overload
def searchsorted(
a: ArrayLike,
v: _ScalarLike_co,
side: _SortSide = ...,
sorter: None | _ArrayLikeInt_co = ..., # 1D int array
) -> intp: ...
@overload
def searchsorted(
a: ArrayLike,
v: ArrayLike,
side: _SortSide = ...,
sorter: None | _ArrayLikeInt_co = ..., # 1D int array
) -> NDArray[intp]: ...
@overload
def resize(
a: _ArrayLike[_SCT],
new_shape: _ShapeLike,
) -> NDArray[_SCT]: ...
@overload
def resize(
a: ArrayLike,
new_shape: _ShapeLike,
) -> NDArray[Any]: ...
@overload
def squeeze(
a: _SCT,
axis: None | _ShapeLike = ...,
) -> _SCT: ...
@overload
def squeeze(
a: _ArrayLike[_SCT],
axis: None | _ShapeLike = ...,
) -> NDArray[_SCT]: ...
@overload
def squeeze(
a: ArrayLike,
axis: None | _ShapeLike = ...,
) -> NDArray[Any]: ...
@overload
def diagonal(
a: _ArrayLike[_SCT],
offset: SupportsIndex = ...,
axis1: SupportsIndex = ...,
axis2: SupportsIndex = ..., # >= 2D array
) -> NDArray[_SCT]: ...
@overload
def diagonal(
a: ArrayLike,
offset: SupportsIndex = ...,
axis1: SupportsIndex = ...,
axis2: SupportsIndex = ..., # >= 2D array
) -> NDArray[Any]: ...
@overload
def trace(
a: ArrayLike, # >= 2D array
offset: SupportsIndex = ...,
axis1: SupportsIndex = ...,
axis2: SupportsIndex = ...,
dtype: DTypeLike = ...,
out: None = ...,
) -> Any: ...
@overload
def trace(
a: ArrayLike, # >= 2D array
offset: SupportsIndex = ...,
axis1: SupportsIndex = ...,
axis2: SupportsIndex = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
) -> _ArrayType: ...
@overload
def ravel(a: _ArrayLike[_SCT], order: _OrderKACF = ...) -> NDArray[_SCT]: ...
@overload
def ravel(a: ArrayLike, order: _OrderKACF = ...) -> NDArray[Any]: ...
def nonzero(a: ArrayLike) -> tuple[NDArray[intp], ...]: ...
def shape(a: ArrayLike) -> _Shape: ...
@overload
def compress(
condition: _ArrayLikeBool_co, # 1D bool array
a: _ArrayLike[_SCT],
axis: None | SupportsIndex = ...,
out: None = ...,
) -> NDArray[_SCT]: ...
@overload
def compress(
condition: _ArrayLikeBool_co, # 1D bool array
a: ArrayLike,
axis: None | SupportsIndex = ...,
out: None = ...,
) -> NDArray[Any]: ...
@overload
def compress(
condition: _ArrayLikeBool_co, # 1D bool array
a: ArrayLike,
axis: None | SupportsIndex = ...,
out: _ArrayType = ...,
) -> _ArrayType: ...
@overload
def clip(
a: _SCT,
a_min: None | ArrayLike,
a_max: None | ArrayLike,
out: None = ...,
*,
min: None | ArrayLike = ...,
max: None | ArrayLike = ...,
dtype: None = ...,
where: None | _ArrayLikeBool_co = ...,
order: _OrderKACF = ...,
subok: bool = ...,
signature: str | tuple[None | str, ...] = ...,
casting: _CastingKind = ...,
) -> _SCT: ...
@overload
def clip(
a: _ScalarLike_co,
a_min: None | ArrayLike,
a_max: None | ArrayLike,
out: None = ...,
*,
min: None | ArrayLike = ...,
max: None | ArrayLike = ...,
dtype: None = ...,
where: None | _ArrayLikeBool_co = ...,
order: _OrderKACF = ...,
subok: bool = ...,
signature: str | tuple[None | str, ...] = ...,
casting: _CastingKind = ...,
) -> Any: ...
@overload
def clip(
a: _ArrayLike[_SCT],
a_min: None | ArrayLike,
a_max: None | ArrayLike,
out: None = ...,
*,
min: None | ArrayLike = ...,
max: None | ArrayLike = ...,
dtype: None = ...,
where: None | _ArrayLikeBool_co = ...,
order: _OrderKACF = ...,
subok: bool = ...,
signature: str | tuple[None | str, ...] = ...,
casting: _CastingKind = ...,
) -> NDArray[_SCT]: ...
@overload
def clip(
a: ArrayLike,
a_min: None | ArrayLike,
a_max: None | ArrayLike,
out: None = ...,
*,
min: None | ArrayLike = ...,
max: None | ArrayLike = ...,
dtype: None = ...,
where: None | _ArrayLikeBool_co = ...,
order: _OrderKACF = ...,
subok: bool = ...,
signature: str | tuple[None | str, ...] = ...,
casting: _CastingKind = ...,
) -> NDArray[Any]: ...
@overload
def clip(
a: ArrayLike,
a_min: None | ArrayLike,
a_max: None | ArrayLike,
out: _ArrayType = ...,
*,
min: None | ArrayLike = ...,
max: None | ArrayLike = ...,
dtype: DTypeLike,
where: None | _ArrayLikeBool_co = ...,
order: _OrderKACF = ...,
subok: bool = ...,
signature: str | tuple[None | str, ...] = ...,
casting: _CastingKind = ...,
) -> Any: ...
@overload
def clip(
a: ArrayLike,
a_min: None | ArrayLike,
a_max: None | ArrayLike,
out: _ArrayType,
*,
min: None | ArrayLike = ...,
max: None | ArrayLike = ...,
dtype: DTypeLike = ...,
where: None | _ArrayLikeBool_co = ...,
order: _OrderKACF = ...,
subok: bool = ...,
signature: str | tuple[None | str, ...] = ...,
casting: _CastingKind = ...,
) -> _ArrayType: ...
@overload
def sum(
a: _ArrayLike[_SCT],
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _SCT: ...
@overload
def sum(
a: ArrayLike,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: None = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def sum(
a: ArrayLike,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _ArrayType: ...
@overload
def all(
a: ArrayLike,
axis: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> np.bool: ...
@overload
def all(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: None = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def all(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> _ArrayType: ...
@overload
def any(
a: ArrayLike,
axis: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> np.bool: ...
@overload
def any(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: None = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def any(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> _ArrayType: ...
@overload
def cumsum(
a: _ArrayLike[_SCT],
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[_SCT]: ...
@overload
def cumsum(
a: ArrayLike,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[Any]: ...
@overload
def cumsum(
a: ArrayLike,
axis: None | SupportsIndex = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
) -> NDArray[_SCT]: ...
@overload
def cumsum(
a: ArrayLike,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: None = ...,
) -> NDArray[Any]: ...
@overload
def cumsum(
a: ArrayLike,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
) -> _ArrayType: ...
@overload
def cumulative_sum(
x: _ArrayLike[_SCT],
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[_SCT]: ...
@overload
def cumulative_sum(
x: ArrayLike,
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[Any]: ...
@overload
def cumulative_sum(
x: ArrayLike,
/,
*,
axis: None | SupportsIndex = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[_SCT]: ...
@overload
def cumulative_sum(
x: ArrayLike,
/,
*,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[Any]: ...
@overload
def cumulative_sum(
x: ArrayLike,
/,
*,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
include_initial: bool = ...,
) -> _ArrayType: ...
@overload
def ptp(
a: _ArrayLike[_SCT],
axis: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
) -> _SCT: ...
@overload
def ptp(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: None = ...,
keepdims: bool = ...,
) -> Any: ...
@overload
def ptp(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
) -> _ArrayType: ...
@overload
def amax(
a: _ArrayLike[_SCT],
axis: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _SCT: ...
@overload
def amax(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: None = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def amax(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _ArrayType: ...
@overload
def amin(
a: _ArrayLike[_SCT],
axis: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _SCT: ...
@overload
def amin(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: None = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def amin(
a: ArrayLike,
axis: None | _ShapeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _ArrayType: ...
# TODO: `np.prod()``: For object arrays `initial` does not necessarily
# have to be a numerical scalar.
# The only requirement is that it is compatible
# with the `.__mul__()` method(s) of the passed array's elements.
# Note that the same situation holds for all wrappers around
# `np.ufunc.reduce`, e.g. `np.sum()` (`.__add__()`).
@overload
def prod(
a: _ArrayLikeBool_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> int_: ...
@overload
def prod(
a: _ArrayLikeUInt_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> uint64: ...
@overload
def prod(
a: _ArrayLikeInt_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> int64: ...
@overload
def prod(
a: _ArrayLikeFloat_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> floating[Any]: ...
@overload
def prod(
a: _ArrayLikeComplex_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> complexfloating[Any, Any]: ...
@overload
def prod(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: None = ...,
out: None = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def prod(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
keepdims: Literal[False] = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _SCT: ...
@overload
def prod(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: None | DTypeLike = ...,
out: None = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def prod(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: None | DTypeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
initial: _NumberLike_co = ...,
where: _ArrayLikeBool_co = ...,
) -> _ArrayType: ...
@overload
def cumprod(
a: _ArrayLikeBool_co,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[int_]: ...
@overload
def cumprod(
a: _ArrayLikeUInt_co,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[uint64]: ...
@overload
def cumprod(
a: _ArrayLikeInt_co,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[int64]: ...
@overload
def cumprod(
a: _ArrayLikeFloat_co,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[floating[Any]]: ...
@overload
def cumprod(
a: _ArrayLikeComplex_co,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[complexfloating[Any, Any]]: ...
@overload
def cumprod(
a: _ArrayLikeObject_co,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
) -> NDArray[object_]: ...
@overload
def cumprod(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | SupportsIndex = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
) -> NDArray[_SCT]: ...
@overload
def cumprod(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: None = ...,
) -> NDArray[Any]: ...
@overload
def cumprod(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
) -> _ArrayType: ...
@overload
def cumulative_prod(
x: _ArrayLikeBool_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[int_]: ...
@overload
def cumulative_prod(
x: _ArrayLikeUInt_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[uint64]: ...
@overload
def cumulative_prod(
x: _ArrayLikeInt_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[int64]: ...
@overload
def cumulative_prod(
x: _ArrayLikeFloat_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[floating[Any]]: ...
@overload
def cumulative_prod(
x: _ArrayLikeComplex_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[complexfloating[Any, Any]]: ...
@overload
def cumulative_prod(
x: _ArrayLikeObject_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: None = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[object_]: ...
@overload
def cumulative_prod(
x: _ArrayLikeComplex_co | _ArrayLikeObject_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[_SCT]: ...
@overload
def cumulative_prod(
x: _ArrayLikeComplex_co | _ArrayLikeObject_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: None = ...,
include_initial: bool = ...,
) -> NDArray[Any]: ...
@overload
def cumulative_prod(
x: _ArrayLikeComplex_co | _ArrayLikeObject_co,
/,
*,
axis: None | SupportsIndex = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
include_initial: bool = ...,
) -> _ArrayType: ...
def ndim(a: ArrayLike) -> int: ...
def size(a: ArrayLike, axis: None | int = ...) -> int: ...
@overload
def around(
a: _BoolLike_co,
decimals: SupportsIndex = ...,
out: None = ...,
) -> float16: ...
@overload
def around(
a: _SCT_uifcO,
decimals: SupportsIndex = ...,
out: None = ...,
) -> _SCT_uifcO: ...
@overload
def around(
a: _ComplexLike_co | object_,
decimals: SupportsIndex = ...,
out: None = ...,
) -> Any: ...
@overload
def around(
a: _ArrayLikeBool_co,
decimals: SupportsIndex = ...,
out: None = ...,
) -> NDArray[float16]: ...
@overload
def around(
a: _ArrayLike[_SCT_uifcO],
decimals: SupportsIndex = ...,
out: None = ...,
) -> NDArray[_SCT_uifcO]: ...
@overload
def around(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
decimals: SupportsIndex = ...,
out: None = ...,
) -> NDArray[Any]: ...
@overload
def around(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
decimals: SupportsIndex = ...,
out: _ArrayType = ...,
) -> _ArrayType: ...
@overload
def mean(
a: _ArrayLikeFloat_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> floating[Any]: ...
@overload
def mean(
a: _ArrayLikeComplex_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> complexfloating[Any, Any]: ...
@overload
def mean(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: None = ...,
out: None = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def mean(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> _SCT: ...
@overload
def mean(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: None = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> Any: ...
@overload
def mean(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
) -> _ArrayType: ...
@overload
def std(
a: _ArrayLikeComplex_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co = ...,
correction: int | float = ...,
) -> floating[Any]: ...
@overload
def std(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: None = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> Any: ...
@overload
def std(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> _SCT: ...
@overload
def std(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> Any: ...
@overload
def std(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
ddof: int | float = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> _ArrayType: ...
@overload
def var(
a: _ArrayLikeComplex_co,
axis: None = ...,
dtype: None = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co = ...,
correction: int | float = ...,
) -> floating[Any]: ...
@overload
def var(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: None = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> Any: ...
@overload
def var(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None = ...,
dtype: _DTypeLike[_SCT] = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: Literal[False] = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> _SCT: ...
@overload
def var(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: None = ...,
ddof: int | float = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> Any: ...
@overload
def var(
a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
axis: None | _ShapeLike = ...,
dtype: DTypeLike = ...,
out: _ArrayType = ...,
ddof: int | float = ...,
keepdims: bool = ...,
*,
where: _ArrayLikeBool_co = ...,
mean: _ArrayLikeComplex_co | _ArrayLikeObject_co = ...,
correction: int | float = ...,
) -> _ArrayType: ...
max = amax
min = amin
round = around