| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060 |
- import datetime as dt
- from collections.abc import Sequence
- from typing import Union, Any, overload, TypeVar, Literal, SupportsIndex
- from numpy import (
- ndarray,
- number,
- uint64,
- int_,
- int64,
- intp,
- float16,
- bool_,
- 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],
- newshape: _ShapeLike,
- order: _OrderACF = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def reshape(
- a: ArrayLike,
- newshape: _ShapeLike,
- order: _OrderACF = ...,
- ) -> 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 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] = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def sort(
- a: ArrayLike,
- axis: None | SupportsIndex = ...,
- kind: None | _SortKind = ...,
- order: None | str | Sequence[str] = ...,
- ) -> NDArray[Any]: ...
- def argsort(
- a: ArrayLike,
- axis: None | SupportsIndex = ...,
- kind: None | _SortKind = ...,
- order: None | str | Sequence[str] = ...,
- ) -> 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 = ...,
- *,
- dtype: None = ...,
- where: None | _ArrayLikeBool_co = ...,
- order: _OrderKACF = ...,
- subok: bool = ...,
- signature: str | tuple[None | str, ...] = ...,
- extobj: list[Any] = ...,
- casting: _CastingKind = ...,
- ) -> _SCT: ...
- @overload
- def clip(
- a: _ScalarLike_co,
- a_min: None | ArrayLike,
- a_max: None | ArrayLike,
- out: None = ...,
- *,
- dtype: None = ...,
- where: None | _ArrayLikeBool_co = ...,
- order: _OrderKACF = ...,
- subok: bool = ...,
- signature: str | tuple[None | str, ...] = ...,
- extobj: list[Any] = ...,
- casting: _CastingKind = ...,
- ) -> Any: ...
- @overload
- def clip(
- a: _ArrayLike[_SCT],
- a_min: None | ArrayLike,
- a_max: None | ArrayLike,
- out: None = ...,
- *,
- dtype: None = ...,
- where: None | _ArrayLikeBool_co = ...,
- order: _OrderKACF = ...,
- subok: bool = ...,
- signature: str | tuple[None | str, ...] = ...,
- extobj: list[Any] = ...,
- casting: _CastingKind = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def clip(
- a: ArrayLike,
- a_min: None | ArrayLike,
- a_max: None | ArrayLike,
- out: None = ...,
- *,
- dtype: None = ...,
- where: None | _ArrayLikeBool_co = ...,
- order: _OrderKACF = ...,
- subok: bool = ...,
- signature: str | tuple[None | str, ...] = ...,
- extobj: list[Any] = ...,
- casting: _CastingKind = ...,
- ) -> NDArray[Any]: ...
- @overload
- def clip(
- a: ArrayLike,
- a_min: None | ArrayLike,
- a_max: None | ArrayLike,
- out: _ArrayType = ...,
- *,
- dtype: DTypeLike,
- where: None | _ArrayLikeBool_co = ...,
- order: _OrderKACF = ...,
- subok: bool = ...,
- signature: str | tuple[None | str, ...] = ...,
- extobj: list[Any] = ...,
- casting: _CastingKind = ...,
- ) -> Any: ...
- @overload
- def clip(
- a: ArrayLike,
- a_min: None | ArrayLike,
- a_max: None | ArrayLike,
- out: _ArrayType,
- *,
- dtype: DTypeLike = ...,
- where: None | _ArrayLikeBool_co = ...,
- order: _OrderKACF = ...,
- subok: bool = ...,
- signature: str | tuple[None | str, ...] = ...,
- extobj: list[Any] = ...,
- 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 = ...,
- ) -> 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 = ...,
- ) -> 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 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: ...
- 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: float = ...,
- keepdims: Literal[False] = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> floating[Any]: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None | _ShapeLike = ...,
- dtype: None = ...,
- out: None = ...,
- ddof: float = ...,
- keepdims: bool = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> Any: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None = ...,
- dtype: _DTypeLike[_SCT] = ...,
- out: None = ...,
- ddof: float = ...,
- keepdims: Literal[False] = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> _SCT: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None | _ShapeLike = ...,
- dtype: DTypeLike = ...,
- out: None = ...,
- ddof: float = ...,
- keepdims: bool = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> Any: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None | _ShapeLike = ...,
- dtype: DTypeLike = ...,
- out: _ArrayType = ...,
- ddof: float = ...,
- keepdims: bool = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> _ArrayType: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co,
- axis: None = ...,
- dtype: None = ...,
- out: None = ...,
- ddof: float = ...,
- keepdims: Literal[False] = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> floating[Any]: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None | _ShapeLike = ...,
- dtype: None = ...,
- out: None = ...,
- ddof: float = ...,
- keepdims: bool = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> Any: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None = ...,
- dtype: _DTypeLike[_SCT] = ...,
- out: None = ...,
- ddof: float = ...,
- keepdims: Literal[False] = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> _SCT: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None | _ShapeLike = ...,
- dtype: DTypeLike = ...,
- out: None = ...,
- ddof: float = ...,
- keepdims: bool = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> Any: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None | _ShapeLike = ...,
- dtype: DTypeLike = ...,
- out: _ArrayType = ...,
- ddof: float = ...,
- keepdims: bool = ...,
- *,
- where: _ArrayLikeBool_co = ...,
- ) -> _ArrayType: ...
- max = amax
- min = amin
- round = around
|