Python NumPy digitize方法用法及代碼示例

例子

基本用法

考慮以下代碼片段：

# Our array of values
a = [3, 6.5, 9]
# Our bins
bins = [5, 6, 7, 8]
np.digitize(a, bins)



array([0, 2, 4])

讓我們理解這裏的輸出。

• 第一個值 3 介於 3 <= 5(第一個 bin)之間，因此返回的整數索引為 0。

• 第二個值 6.5 介於 6 和 7 之間(第 2 個和第 3 個 bin)，因此返回的整數索引為 2。

• 第三個值 9 大於 8(第 4 個 bin)，因此返回的整數索引為 4。

解決這個問題的一個好方法是考慮該值的索引(如果要將其插入到 bin 數組中)。例如，值 3 將被插入到索引 0 中，因此返回 0。 6.5 將插入到索引 2 中，因此返回 2，依此類推。

處理端點

默認情況下，當檢查將值放入哪個 bin 時，Numpy 將使用 < 比較。例如，

a = [5]
bins = [5, 6]
np.digitize(a, bins)   # or right=False



array([1])

我們得到整數索引 1 的原因是我們執行的第一次比較是 5 < 5，其計算結果為 False 。

我們可以執行 <= 比較，而不是 < 比較，如下所示：

a = [5]
bins = [5, 6]
np.digitize(a, bins, right=True)



array([0])

在這裏，我們得到整數索引 0，因為第一個比較 5 <=5 的計算結果為 True 。