codekingpro/portable-devtools

1from collections.abc import Sequence
2from typing import Any, Literal as L, SupportsIndex, TypeAlias
3
4from numpy._typing import ArrayLike, NDArray
5
6__all__ = ["histogram", "histogramdd", "histogram_bin_edges"]
7
8_BinKind: TypeAlias = L[
9    "stone",
10    "auto",
11    "doane",
12    "fd",
13    "rice",
14    "scott",
15    "sqrt",
16    "sturges",
17]
18
19def histogram_bin_edges(
20    a: ArrayLike,
21    bins: _BinKind | SupportsIndex | ArrayLike = 10,
22    range: tuple[float, float] | None = None,
23    weights: ArrayLike | None = None,
24) -> NDArray[Any]: ...
25
26def histogram(
27    a: ArrayLike,
28    bins: _BinKind | SupportsIndex | ArrayLike = 10,
29    range: tuple[float, float] | None = None,
30    density: bool | None = None,
31    weights: ArrayLike | None = None,
32) -> tuple[NDArray[Any], NDArray[Any]]: ...
33
34def histogramdd(
35    sample: ArrayLike,
36    bins: SupportsIndex | ArrayLike = 10,
37    range: Sequence[tuple[float, float]] | None = None,
38    density: bool | None = None,
39    weights: ArrayLike | None = None,
40) -> tuple[NDArray[Any], tuple[NDArray[Any], ...]]: ...
41