| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735 |
- # ruff: noqa: ANN401
- from _typeshed import Incomplete
- from collections.abc import Sequence
- from typing import (
- Any,
- Literal,
- Never,
- Protocol,
- SupportsIndex,
- TypeAlias,
- TypedDict,
- TypeVar,
- Unpack,
- overload,
- type_check_only,
- )
- import numpy as np
- from numpy import (
- _AnyShapeT,
- _CastingKind,
- _ModeKind,
- _OrderACF,
- _OrderKACF,
- _PartitionKind,
- _SortKind,
- _SortSide,
- complexfloating,
- float16,
- floating,
- generic,
- int64,
- int_,
- intp,
- object_,
- timedelta64,
- uint64,
- )
- from numpy._globals import _NoValueType
- from numpy._typing import (
- ArrayLike,
- DTypeLike,
- NDArray,
- _AnyShape,
- _ArrayLike,
- _ArrayLikeBool_co,
- _ArrayLikeComplex_co,
- _ArrayLikeFloat_co,
- _ArrayLikeInt,
- _ArrayLikeInt_co,
- _ArrayLikeObject_co,
- _ArrayLikeUInt_co,
- _BoolLike_co,
- _ComplexLike_co,
- _DTypeLike,
- _IntLike_co,
- _NestedSequence,
- _NumberLike_co,
- _ScalarLike_co,
- _ShapeLike,
- )
- __all__ = [
- "all",
- "amax",
- "amin",
- "any",
- "argmax",
- "argmin",
- "argpartition",
- "argsort",
- "around",
- "choose",
- "clip",
- "compress",
- "cumprod",
- "cumsum",
- "cumulative_prod",
- "cumulative_sum",
- "diagonal",
- "mean",
- "max",
- "min",
- "matrix_transpose",
- "ndim",
- "nonzero",
- "partition",
- "prod",
- "ptp",
- "put",
- "ravel",
- "repeat",
- "reshape",
- "resize",
- "round",
- "searchsorted",
- "shape",
- "size",
- "sort",
- "squeeze",
- "std",
- "sum",
- "swapaxes",
- "take",
- "trace",
- "transpose",
- "var",
- ]
- _ScalarT = TypeVar("_ScalarT", bound=generic)
- _NumberOrObjectT = TypeVar("_NumberOrObjectT", bound=np.number | np.object_)
- _ArrayT = TypeVar("_ArrayT", bound=np.ndarray[Any, Any])
- _ShapeT = TypeVar("_ShapeT", bound=tuple[int, ...])
- _ShapeT_co = TypeVar("_ShapeT_co", bound=tuple[int, ...], covariant=True)
- _BoolOrIntArrayT = TypeVar("_BoolOrIntArrayT", bound=NDArray[np.integer | np.bool])
- @type_check_only
- class _SupportsShape(Protocol[_ShapeT_co]):
- # NOTE: it matters that `self` is positional only
- @property
- def shape(self, /) -> _ShapeT_co: ...
- @type_check_only
- class _UFuncKwargs(TypedDict, total=False):
- where: _ArrayLikeBool_co | None
- order: _OrderKACF
- subok: bool
- signature: str | tuple[str | None, ...]
- casting: _CastingKind
- # a "sequence" that isn't a string, bytes, bytearray, or memoryview
- _T = TypeVar("_T")
- _PyArray: TypeAlias = list[_T] | tuple[_T, ...]
- # `int` also covers `bool`
- _PyScalar: TypeAlias = complex | bytes | str
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def take(
- a: _ArrayLike[_ScalarT],
- indices: _IntLike_co,
- axis: None = None,
- out: None = None,
- mode: _ModeKind = "raise",
- ) -> _ScalarT: ...
- @overload
- def take(
- a: ArrayLike,
- indices: _IntLike_co,
- axis: SupportsIndex | None = None,
- out: None = None,
- mode: _ModeKind = "raise",
- ) -> Any: ...
- @overload
- def take(
- a: _ArrayLike[_ScalarT],
- indices: _ArrayLikeInt_co,
- axis: SupportsIndex | None = None,
- out: None = None,
- mode: _ModeKind = "raise",
- ) -> NDArray[_ScalarT]: ...
- @overload
- def take(
- a: ArrayLike,
- indices: _ArrayLikeInt_co,
- axis: SupportsIndex | None = None,
- out: None = None,
- mode: _ModeKind = "raise",
- ) -> NDArray[Any]: ...
- @overload
- def take(
- a: ArrayLike,
- indices: _ArrayLikeInt_co,
- axis: SupportsIndex | None,
- out: _ArrayT,
- mode: _ModeKind = "raise",
- ) -> _ArrayT: ...
- @overload
- def take(
- a: ArrayLike,
- indices: _ArrayLikeInt_co,
- axis: SupportsIndex | None = None,
- *,
- out: _ArrayT,
- mode: _ModeKind = "raise",
- ) -> _ArrayT: ...
- @overload
- def reshape( # shape: index
- a: _ArrayLike[_ScalarT],
- /,
- shape: SupportsIndex,
- order: _OrderACF = "C",
- *,
- copy: bool | None = None,
- ) -> np.ndarray[tuple[int], np.dtype[_ScalarT]]: ...
- @overload
- def reshape( # shape: (int, ...) @ _AnyShapeT
- a: _ArrayLike[_ScalarT],
- /,
- shape: _AnyShapeT,
- order: _OrderACF = "C",
- *,
- copy: bool | None = None,
- ) -> np.ndarray[_AnyShapeT, np.dtype[_ScalarT]]: ...
- @overload # shape: Sequence[index]
- def reshape(
- a: _ArrayLike[_ScalarT],
- /,
- shape: Sequence[SupportsIndex],
- order: _OrderACF = "C",
- *,
- copy: bool | None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload # shape: index
- def reshape(
- a: ArrayLike,
- /,
- shape: SupportsIndex,
- order: _OrderACF = "C",
- *,
- copy: bool | None = None,
- ) -> np.ndarray[tuple[int], np.dtype]: ...
- @overload
- def reshape( # shape: (int, ...) @ _AnyShapeT
- a: ArrayLike,
- /,
- shape: _AnyShapeT,
- order: _OrderACF = "C",
- *,
- copy: bool | None = None,
- ) -> np.ndarray[_AnyShapeT, np.dtype]: ...
- @overload # shape: Sequence[index]
- def reshape(
- a: ArrayLike,
- /,
- shape: Sequence[SupportsIndex],
- order: _OrderACF = "C",
- *,
- copy: bool | None = None,
- ) -> NDArray[Any]: ...
- @overload
- def choose(
- a: _IntLike_co,
- choices: ArrayLike,
- out: None = None,
- mode: _ModeKind = "raise",
- ) -> Any: ...
- @overload
- def choose(
- a: _ArrayLikeInt_co,
- choices: _ArrayLike[_ScalarT],
- out: None = None,
- mode: _ModeKind = "raise",
- ) -> NDArray[_ScalarT]: ...
- @overload
- def choose(
- a: _ArrayLikeInt_co,
- choices: ArrayLike,
- out: None = None,
- mode: _ModeKind = "raise",
- ) -> NDArray[Any]: ...
- @overload
- def choose(
- a: _ArrayLikeInt_co,
- choices: ArrayLike,
- out: _ArrayT,
- mode: _ModeKind = "raise",
- ) -> _ArrayT: ...
- # keep in sync with `ma.core.repeat`
- @overload
- def repeat(
- a: _ArrayLike[_ScalarT],
- repeats: _ArrayLikeInt_co,
- axis: None = None,
- ) -> np.ndarray[tuple[int], np.dtype[_ScalarT]]: ...
- @overload
- def repeat(
- a: _ArrayLike[_ScalarT],
- repeats: _ArrayLikeInt_co,
- axis: SupportsIndex,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def repeat(
- a: ArrayLike,
- repeats: _ArrayLikeInt_co,
- axis: None = None,
- ) -> np.ndarray[tuple[int], np.dtype[Any]]: ...
- @overload
- def repeat(
- a: ArrayLike,
- repeats: _ArrayLikeInt_co,
- axis: SupportsIndex,
- ) -> NDArray[Any]: ...
- #
- def put(
- a: NDArray[Any],
- ind: _ArrayLikeInt_co,
- v: ArrayLike,
- mode: _ModeKind = "raise",
- ) -> None: ...
- # keep in sync with `ndarray.swapaxes` and `ma.core.swapaxes`
- @overload
- def swapaxes(a: _ArrayT, axis1: SupportsIndex, axis2: SupportsIndex) -> _ArrayT: ...
- @overload
- def swapaxes(a: _ArrayLike[_ScalarT], axis1: SupportsIndex, axis2: SupportsIndex) -> NDArray[_ScalarT]: ...
- @overload
- def swapaxes(a: ArrayLike, axis1: SupportsIndex, axis2: SupportsIndex) -> NDArray[Any]: ...
- @overload
- def transpose(
- a: _ArrayLike[_ScalarT],
- axes: _ShapeLike | None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def transpose(
- a: ArrayLike,
- axes: _ShapeLike | None = None,
- ) -> NDArray[Any]: ...
- @overload
- def matrix_transpose(x: _ArrayLike[_ScalarT], /) -> NDArray[_ScalarT]: ...
- @overload
- def matrix_transpose(x: ArrayLike, /) -> NDArray[Any]: ...
- #
- @overload
- def partition(
- a: _ArrayLike[_ScalarT],
- kth: _ArrayLikeInt,
- axis: SupportsIndex | None = -1,
- kind: _PartitionKind = "introselect",
- order: None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def partition(
- a: _ArrayLike[np.void],
- kth: _ArrayLikeInt,
- axis: SupportsIndex | None = -1,
- kind: _PartitionKind = "introselect",
- order: str | Sequence[str] | None = None,
- ) -> NDArray[np.void]: ...
- @overload
- def partition(
- a: ArrayLike,
- kth: _ArrayLikeInt,
- axis: SupportsIndex | None = -1,
- kind: _PartitionKind = "introselect",
- order: str | Sequence[str] | None = None,
- ) -> NDArray[Any]: ...
- #
- def argpartition(
- a: ArrayLike,
- kth: _ArrayLikeInt,
- axis: SupportsIndex | None = -1,
- kind: _PartitionKind = "introselect",
- order: str | Sequence[str] | None = None,
- ) -> NDArray[intp]: ...
- #
- @overload
- def sort(
- a: _ArrayLike[_ScalarT],
- axis: SupportsIndex | None = -1,
- kind: _SortKind | None = None,
- order: str | Sequence[str] | None = None,
- *,
- stable: bool | None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def sort(
- a: ArrayLike,
- axis: SupportsIndex | None = -1,
- kind: _SortKind | None = None,
- order: str | Sequence[str] | None = None,
- *,
- stable: bool | None = None,
- ) -> NDArray[Any]: ...
- def argsort(
- a: ArrayLike,
- axis: SupportsIndex | None = -1,
- kind: _SortKind | None = None,
- order: str | Sequence[str] | None = None,
- *,
- stable: bool | None = None,
- ) -> NDArray[intp]: ...
- @overload
- def argmax(
- a: ArrayLike,
- axis: None = None,
- out: None = None,
- *,
- keepdims: Literal[False] | _NoValueType = ...,
- ) -> intp: ...
- @overload
- def argmax(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- out: None = None,
- *,
- keepdims: bool | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def argmax(
- a: ArrayLike,
- axis: SupportsIndex | None,
- out: _BoolOrIntArrayT,
- *,
- keepdims: bool | _NoValueType = ...,
- ) -> _BoolOrIntArrayT: ...
- @overload
- def argmax(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- *,
- out: _BoolOrIntArrayT,
- keepdims: bool | _NoValueType = ...,
- ) -> _BoolOrIntArrayT: ...
- @overload
- def argmin(
- a: ArrayLike,
- axis: None = None,
- out: None = None,
- *,
- keepdims: Literal[False] | _NoValueType = ...,
- ) -> intp: ...
- @overload
- def argmin(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- out: None = None,
- *,
- keepdims: bool | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def argmin(
- a: ArrayLike,
- axis: SupportsIndex | None,
- out: _BoolOrIntArrayT,
- *,
- keepdims: bool | _NoValueType = ...,
- ) -> _BoolOrIntArrayT: ...
- @overload
- def argmin(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- *,
- out: _BoolOrIntArrayT,
- keepdims: bool | _NoValueType = ...,
- ) -> _BoolOrIntArrayT: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def searchsorted(
- a: ArrayLike,
- v: _ScalarLike_co,
- side: _SortSide = "left",
- sorter: _ArrayLikeInt_co | None = None, # 1D int array
- ) -> intp: ...
- @overload
- def searchsorted(
- a: ArrayLike,
- v: ArrayLike,
- side: _SortSide = "left",
- sorter: _ArrayLikeInt_co | None = None, # 1D int array
- ) -> NDArray[intp]: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def resize(a: _ArrayLike[_ScalarT], new_shape: SupportsIndex | tuple[SupportsIndex]) -> np.ndarray[tuple[int], np.dtype[_ScalarT]]: ...
- @overload
- def resize(a: _ArrayLike[_ScalarT], new_shape: _AnyShapeT) -> np.ndarray[_AnyShapeT, np.dtype[_ScalarT]]: ...
- @overload
- def resize(a: _ArrayLike[_ScalarT], new_shape: _ShapeLike) -> NDArray[_ScalarT]: ...
- @overload
- def resize(a: ArrayLike, new_shape: SupportsIndex | tuple[SupportsIndex]) -> np.ndarray[tuple[int], np.dtype]: ...
- @overload
- def resize(a: ArrayLike, new_shape: _AnyShapeT) -> np.ndarray[_AnyShapeT, np.dtype]: ...
- @overload
- def resize(a: ArrayLike, new_shape: _ShapeLike) -> NDArray[Any]: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def squeeze(
- a: _ScalarT,
- axis: _ShapeLike | None = None,
- ) -> _ScalarT: ...
- @overload
- def squeeze(
- a: _ArrayLike[_ScalarT],
- axis: _ShapeLike | None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def squeeze(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- ) -> NDArray[Any]: ...
- # keep in sync with `ma.core.diagonal`
- @overload
- def diagonal(
- a: _ArrayLike[_ScalarT],
- offset: SupportsIndex = 0,
- axis1: SupportsIndex = 0,
- axis2: SupportsIndex = 1, # >= 2D array
- ) -> NDArray[_ScalarT]: ...
- @overload
- def diagonal(
- a: ArrayLike,
- offset: SupportsIndex = 0,
- axis1: SupportsIndex = 0,
- axis2: SupportsIndex = 1, # >= 2D array
- ) -> NDArray[Any]: ...
- # keep in sync with `ma.core.trace`
- @overload
- def trace(
- a: ArrayLike, # >= 2D array
- offset: SupportsIndex = 0,
- axis1: SupportsIndex = 0,
- axis2: SupportsIndex = 1,
- dtype: DTypeLike | None = None,
- out: None = None,
- ) -> Any: ...
- @overload
- def trace(
- a: ArrayLike, # >= 2D array
- offset: SupportsIndex,
- axis1: SupportsIndex,
- axis2: SupportsIndex,
- dtype: DTypeLike | None,
- out: _ArrayT,
- ) -> _ArrayT: ...
- @overload
- def trace(
- a: ArrayLike, # >= 2D array
- offset: SupportsIndex = 0,
- axis1: SupportsIndex = 0,
- axis2: SupportsIndex = 1,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- ) -> _ArrayT: ...
- _Array1D: TypeAlias = np.ndarray[tuple[int], np.dtype[_ScalarT]]
- @overload
- def ravel(a: _ArrayLike[_ScalarT], order: _OrderKACF = "C") -> _Array1D[_ScalarT]: ...
- @overload
- def ravel(a: bytes | _NestedSequence[bytes], order: _OrderKACF = "C") -> _Array1D[np.bytes_]: ...
- @overload
- def ravel(a: str | _NestedSequence[str], order: _OrderKACF = "C") -> _Array1D[np.str_]: ...
- @overload
- def ravel(a: bool | _NestedSequence[bool], order: _OrderKACF = "C") -> _Array1D[np.bool]: ...
- @overload
- def ravel(a: int | _NestedSequence[int], order: _OrderKACF = "C") -> _Array1D[np.int_ | Any]: ...
- @overload
- def ravel(a: float | _NestedSequence[float], order: _OrderKACF = "C") -> _Array1D[np.float64 | Any]: ...
- @overload
- def ravel(a: complex | _NestedSequence[complex], order: _OrderKACF = "C") -> _Array1D[np.complex128 | Any]: ...
- @overload
- def ravel(a: ArrayLike, order: _OrderKACF = "C") -> np.ndarray[tuple[int], np.dtype]: ...
- def nonzero(a: _ArrayLike[Any]) -> tuple[np.ndarray[tuple[int], np.dtype[intp]], ...]: ...
- # this prevents `Any` from being returned with Pyright
- @overload
- def shape(a: _SupportsShape[Never]) -> _AnyShape: ...
- @overload
- def shape(a: _SupportsShape[_ShapeT]) -> _ShapeT: ...
- @overload
- def shape(a: _PyScalar) -> tuple[()]: ...
- # `collections.abc.Sequence` can't be used hesre, since `bytes` and `str` are
- # subtypes of it, which would make the return types incompatible.
- @overload
- def shape(a: _PyArray[_PyScalar]) -> tuple[int]: ...
- @overload
- def shape(a: _PyArray[_PyArray[_PyScalar]]) -> tuple[int, int]: ...
- # this overload will be skipped by typecheckers that don't support PEP 688
- @overload
- def shape(a: memoryview | bytearray) -> tuple[int]: ...
- @overload
- def shape(a: ArrayLike) -> _AnyShape: ...
- @overload
- def compress(
- condition: _ArrayLikeBool_co, # 1D bool array
- a: _ArrayLike[_ScalarT],
- axis: SupportsIndex | None = None,
- out: None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def compress(
- condition: _ArrayLikeBool_co, # 1D bool array
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- out: None = None,
- ) -> NDArray[Any]: ...
- @overload
- def compress(
- condition: _ArrayLikeBool_co, # 1D bool array
- a: ArrayLike,
- axis: SupportsIndex | None,
- out: _ArrayT,
- ) -> _ArrayT: ...
- @overload
- def compress(
- condition: _ArrayLikeBool_co, # 1D bool array
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- *,
- out: _ArrayT,
- ) -> _ArrayT: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def clip(
- a: _ScalarT,
- a_min: ArrayLike | _NoValueType | None = ...,
- a_max: ArrayLike | _NoValueType | None = ...,
- out: None = None,
- *,
- min: ArrayLike | _NoValueType | None = ...,
- max: ArrayLike | _NoValueType | None = ...,
- dtype: None = None,
- **kwargs: Unpack[_UFuncKwargs],
- ) -> _ScalarT: ...
- @overload
- def clip(
- a: _ScalarLike_co,
- a_min: ArrayLike | _NoValueType | None = ...,
- a_max: ArrayLike | _NoValueType | None = ...,
- out: None = None,
- *,
- min: ArrayLike | _NoValueType | None = ...,
- max: ArrayLike | _NoValueType | None = ...,
- dtype: None = None,
- **kwargs: Unpack[_UFuncKwargs],
- ) -> Any: ...
- @overload
- def clip(
- a: _ArrayLike[_ScalarT],
- a_min: ArrayLike | _NoValueType | None = ...,
- a_max: ArrayLike | _NoValueType | None = ...,
- out: None = None,
- *,
- min: ArrayLike | _NoValueType | None = ...,
- max: ArrayLike | _NoValueType | None = ...,
- dtype: None = None,
- **kwargs: Unpack[_UFuncKwargs],
- ) -> NDArray[_ScalarT]: ...
- @overload
- def clip(
- a: ArrayLike,
- a_min: ArrayLike | _NoValueType | None = ...,
- a_max: ArrayLike | _NoValueType | None = ...,
- out: None = None,
- *,
- min: ArrayLike | _NoValueType | None = ...,
- max: ArrayLike | _NoValueType | None = ...,
- dtype: None = None,
- **kwargs: Unpack[_UFuncKwargs],
- ) -> NDArray[Any]: ...
- @overload
- def clip(
- a: ArrayLike,
- a_min: ArrayLike | None,
- a_max: ArrayLike | None,
- out: _ArrayT,
- *,
- min: ArrayLike | _NoValueType | None = ...,
- max: ArrayLike | _NoValueType | None = ...,
- dtype: DTypeLike | None = None,
- **kwargs: Unpack[_UFuncKwargs],
- ) -> _ArrayT: ...
- @overload
- def clip(
- a: ArrayLike,
- a_min: ArrayLike | _NoValueType | None = ...,
- a_max: ArrayLike | _NoValueType | None = ...,
- *,
- out: _ArrayT,
- min: ArrayLike | _NoValueType | None = ...,
- max: ArrayLike | _NoValueType | None = ...,
- dtype: DTypeLike | None = None,
- **kwargs: Unpack[_UFuncKwargs],
- ) -> _ArrayT: ...
- @overload
- def clip(
- a: ArrayLike,
- a_min: ArrayLike | _NoValueType | None = ...,
- a_max: ArrayLike | _NoValueType | None = ...,
- out: None = None,
- *,
- min: ArrayLike | _NoValueType | None = ...,
- max: ArrayLike | _NoValueType | None = ...,
- dtype: DTypeLike | None = None,
- **kwargs: Unpack[_UFuncKwargs],
- ) -> Any: ...
- @overload
- def sum(
- a: _ArrayLike[_ScalarT],
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def sum(
- a: _ArrayLike[_ScalarT],
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT | NDArray[_ScalarT]: ...
- @overload
- def sum(
- a: ArrayLike,
- axis: None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def sum(
- a: ArrayLike,
- axis: None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def sum(
- a: ArrayLike,
- axis: _ShapeLike | None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT | NDArray[_ScalarT]: ...
- @overload
- def sum(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT | NDArray[_ScalarT]: ...
- @overload
- def sum(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def sum(
- a: ArrayLike,
- axis: _ShapeLike | None,
- dtype: DTypeLike | None,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def sum(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- # keep in sync with `any`
- @overload
- def all(
- a: ArrayLike | None,
- axis: None = None,
- out: None = None,
- keepdims: Literal[False, 0] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> np.bool: ...
- @overload
- def all(
- a: ArrayLike | None,
- axis: int | tuple[int, ...] | None = None,
- out: None = None,
- keepdims: _BoolLike_co | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Incomplete: ...
- @overload
- def all(
- a: ArrayLike | None,
- axis: int | tuple[int, ...] | None,
- out: _ArrayT,
- keepdims: _BoolLike_co | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def all(
- a: ArrayLike | None,
- axis: int | tuple[int, ...] | None = None,
- *,
- out: _ArrayT,
- keepdims: _BoolLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- # keep in sync with `all`
- @overload
- def any(
- a: ArrayLike | None,
- axis: None = None,
- out: None = None,
- keepdims: Literal[False, 0] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> np.bool: ...
- @overload
- def any(
- a: ArrayLike | None,
- axis: int | tuple[int, ...] | None = None,
- out: None = None,
- keepdims: _BoolLike_co | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Incomplete: ...
- @overload
- def any(
- a: ArrayLike | None,
- axis: int | tuple[int, ...] | None,
- out: _ArrayT,
- keepdims: _BoolLike_co | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def any(
- a: ArrayLike | None,
- axis: int | tuple[int, ...] | None = None,
- *,
- out: _ArrayT,
- keepdims: _BoolLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- #
- @overload
- def cumsum(
- a: _ArrayLike[_ScalarT],
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumsum(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[Any]: ...
- @overload
- def cumsum(
- a: ArrayLike,
- axis: SupportsIndex | None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumsum(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumsum(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- ) -> NDArray[Any]: ...
- @overload
- def cumsum(
- a: ArrayLike,
- axis: SupportsIndex | None,
- dtype: DTypeLike | None,
- out: _ArrayT,
- ) -> _ArrayT: ...
- @overload
- def cumsum(
- a: ArrayLike,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- ) -> _ArrayT: ...
- @overload
- def cumulative_sum(
- x: _ArrayLike[_ScalarT],
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumulative_sum(
- x: ArrayLike,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[Any]: ...
- @overload
- def cumulative_sum(
- x: ArrayLike,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumulative_sum(
- x: ArrayLike,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[Any]: ...
- @overload
- def cumulative_sum(
- x: ArrayLike,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- out: _ArrayT,
- include_initial: bool = False,
- ) -> _ArrayT: ...
- @overload
- def ptp(
- a: _ArrayLike[_ScalarT],
- axis: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def ptp(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def ptp(
- a: ArrayLike,
- axis: _ShapeLike | None,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def ptp(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- *,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def amax(
- a: _ArrayLike[_ScalarT],
- axis: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def amax(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def amax(
- a: ArrayLike,
- axis: _ShapeLike | None,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def amax(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- *,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def amin(
- a: _ArrayLike[_ScalarT],
- axis: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def amin(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def amin(
- a: ArrayLike,
- axis: _ShapeLike | None,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def amin(
- a: ArrayLike,
- axis: _ShapeLike | None = None,
- *,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- # 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__()`).
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def prod(
- a: _ArrayLikeBool_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> int_: ...
- @overload
- def prod(
- a: _ArrayLikeUInt_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> uint64: ...
- @overload
- def prod(
- a: _ArrayLikeInt_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> int64: ...
- @overload
- def prod(
- a: _ArrayLikeFloat_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> floating: ...
- @overload
- def prod(
- a: _ArrayLikeComplex_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> complexfloating: ...
- @overload
- def prod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def prod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def prod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def prod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def prod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None,
- dtype: DTypeLike | None,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def prod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- initial: _NumberLike_co | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def cumprod(
- a: _ArrayLikeBool_co,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[int_]: ...
- @overload
- def cumprod(
- a: _ArrayLikeUInt_co,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[uint64]: ...
- @overload
- def cumprod(
- a: _ArrayLikeInt_co,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[int64]: ...
- @overload
- def cumprod(
- a: _ArrayLikeFloat_co,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[floating]: ...
- @overload
- def cumprod(
- a: _ArrayLikeComplex_co,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[complexfloating]: ...
- @overload
- def cumprod(
- a: _ArrayLikeObject_co,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- ) -> NDArray[object_]: ...
- @overload
- def cumprod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: SupportsIndex | None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumprod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: SupportsIndex | None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumprod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- ) -> NDArray[Any]: ...
- @overload
- def cumprod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: SupportsIndex | None,
- dtype: DTypeLike | None,
- out: _ArrayT,
- ) -> _ArrayT: ...
- @overload
- def cumprod(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- ) -> _ArrayT: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def cumulative_prod(
- x: _ArrayLikeBool_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[int_]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeUInt_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[uint64]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeInt_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[int64]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeFloat_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[floating]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeComplex_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[complexfloating]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeObject_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[object_]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- include_initial: bool = False,
- ) -> NDArray[Any]: ...
- @overload
- def cumulative_prod(
- x: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- /,
- *,
- axis: SupportsIndex | None = None,
- dtype: DTypeLike | None = None,
- out: _ArrayT,
- include_initial: bool = False,
- ) -> _ArrayT: ...
- def ndim(a: ArrayLike) -> int: ...
- def size(a: ArrayLike, axis: int | tuple[int, ...] | None = None) -> int: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def around(
- a: _BoolLike_co,
- decimals: SupportsIndex = 0,
- out: None = None,
- ) -> float16: ...
- @overload
- def around(
- a: _NumberOrObjectT,
- decimals: SupportsIndex = 0,
- out: None = None,
- ) -> _NumberOrObjectT: ...
- @overload
- def around(
- a: _ComplexLike_co | object_,
- decimals: SupportsIndex = 0,
- out: None = None,
- ) -> Any: ...
- @overload
- def around(
- a: _ArrayLikeBool_co,
- decimals: SupportsIndex = 0,
- out: None = None,
- ) -> NDArray[float16]: ...
- @overload
- def around(
- a: _ArrayLike[_NumberOrObjectT],
- decimals: SupportsIndex = 0,
- out: None = None,
- ) -> NDArray[_NumberOrObjectT]: ...
- @overload
- def around(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- decimals: SupportsIndex = 0,
- out: None = None,
- ) -> NDArray[Any]: ...
- @overload
- def around(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- decimals: SupportsIndex,
- out: _ArrayT,
- ) -> _ArrayT: ...
- @overload
- def around(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- decimals: SupportsIndex = 0,
- *,
- out: _ArrayT,
- ) -> _ArrayT: ...
- # TODO: Fix overlapping overloads: https://github.com/numpy/numpy/issues/27032
- @overload
- def mean(
- a: _ArrayLikeFloat_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> floating: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> complexfloating: ...
- @overload
- def mean(
- a: _ArrayLike[np.timedelta64],
- axis: None = None,
- dtype: None = None,
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> timedelta64: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None,
- dtype: DTypeLike | None,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- keepdims: bool | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: Literal[False] | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None,
- dtype: _DTypeLike[_ScalarT],
- out: None,
- keepdims: Literal[True, 1],
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> NDArray[_ScalarT]: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- *,
- keepdims: bool | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT | NDArray[_ScalarT]: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> _ScalarT | NDArray[_ScalarT]: ...
- @overload
- def mean(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- ) -> Incomplete: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- ddof: float = 0,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> floating: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: None = None,
- out: None = None,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ddof: float = 0,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ddof: float = 0,
- keepdims: Literal[False] | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None,
- dtype: DTypeLike | None,
- out: _ArrayT,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def std(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co,
- axis: None = None,
- dtype: None = None,
- out: None = None,
- ddof: float = 0,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> floating: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: None = None,
- out: None = None,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ddof: float = 0,
- keepdims: Literal[False] | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: None = None,
- *,
- dtype: _DTypeLike[_ScalarT],
- out: None = None,
- ddof: float = 0,
- keepdims: Literal[False] | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ScalarT: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- out: None = None,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> Any: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None,
- dtype: DTypeLike | None,
- out: _ArrayT,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- *,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ArrayT: ...
- @overload
- def var(
- a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
- axis: _ShapeLike | None = None,
- dtype: DTypeLike | None = None,
- *,
- out: _ArrayT,
- ddof: float = 0,
- keepdims: bool | _NoValueType = ...,
- where: _ArrayLikeBool_co | _NoValueType = ...,
- mean: _ArrayLikeComplex_co | _ArrayLikeObject_co | _NoValueType = ...,
- correction: float | _NoValueType = ...,
- ) -> _ArrayT: ...
- max = amax
- min = amin
- round = around
|