jax.numpy.histogram

内容

jax.numpy.histogram#

jax.numpy.histogram(a, bins=10, range=None, weights=None, density=None)[source]#

计算数据集的直方图。

LAX 后端实现的 numpy.histogram().

原始文档字符串如下。

参数:
  • a (array_like) – 输入数据。直方图是在扁平化的数组上计算的。

  • bins (intsequence of scalarsstr, 可选) – 如果 `bins` 是一个整数,它定义了给定范围内等宽箱子的数量(默认值为 10)。 如果 `bins` 是一个序列,它定义了一个单调递增的箱边数组,包括最右边的边,允许非均匀的箱宽度。

  • 范围 ((浮点数, 浮点数), 可选) – 箱子的上下限范围。如果未提供,则范围仅为 (a.min(), a.max())。范围之外的值将被忽略。范围的第一个元素必须小于或等于第二个元素。 范围 也会影响自动箱体计算。虽然箱体宽度是基于 范围 内的实际数据计算出的最佳值,但箱体数量会填满整个范围,包括不包含数据的部分。

  • 权重 (类数组, 可选) – 与 a 形状相同的权重数组。a 中的每个值仅对其关联的权重(而不是 1)贡献到箱体计数。如果 密度 为 True,则权重会标准化,因此密度在范围内的积分保持为 1。请注意,dtype权重 也会成为返回累加器 (hist) 的 dtype,因此它必须足够大以容纳累加的值。

  • 密度 (布尔值, 可选) – 如果 False,结果将包含每个箱体中的样本数量。如果 True,结果是箱体内概率密度函数的值,标准化后,积分在范围内的值为 1。请注意,直方图值的总和将不会等于 1,除非选择单位宽度箱体;它不是概率质量函数。

返回值:

  • hist (数组) – 直方图的值。有关可能语义的描述,请参阅 密度权重。如果给出 权重,则 hist.dtype 将取自 权重

  • bin_edges (dtype 为浮点数的数组) – 返回箱体边沿 (length(hist)+1)

返回类型:

元组[数组, 数组]