numpy.digitize函数介绍
numpy.digitize
是 NumPy 中的一个函数,用于根据给定的边界(分箱规则)对输入数组中的值进行分箱,返回对应的箱索引。它在分段处理、统计分布和离散化数据等场景中非常有用。
函数签名
numpy.digitize(x, bins, right=False)
参数详解
-
x
:- 输入数组,表示需要进行分箱的值。
- 可以是一维或多维数组,最终结果的形状与
x
相同。
-
bins
:- 一维数组,表示分箱的边界。
- 必须是 单调递增 或 单调递减 的。
-
right
(可选):- 布尔值,默认为
False
。 - 指定是否使用右边界作为开区间:
right=False
:左闭右开区间 [bins[i−1],bins[i])。right=True
:左开右闭区间 (bins[i−1],bins[i]]。
- 布尔值,默认为
返回值
- 一个数组,与
x
形状相同,表示输入值所属的箱索引。 - 索引范围:
- 如果 x<bins