溫馨提示×

NumPy怎么將連續(xù)值轉(zhuǎn)換為離散值

小億
99
2024-05-13 12:54:14
欄目: 編程語言

NumPy可以使用numpy.digitize函數(shù)將連續(xù)值轉(zhuǎn)換為離散值。numpy.digitize函數(shù)接受兩個參數(shù)，第一個參數(shù)為要轉(zhuǎn)換的連續(xù)值，第二個參數(shù)為用來劃分連續(xù)值的邊界數(shù)組。函數(shù)返回一個整數(shù)數(shù)組，表示每個連續(xù)值所屬的離散區(qū)間索引。

以下是一個示例代碼，將連續(xù)值轉(zhuǎn)換為離散值：

import numpy as np

# 創(chuàng)建連續(xù)值數(shù)組
values = np.array([1.5, 2.5, 3.5, 4.5, 5.5])

# 創(chuàng)建邊界數(shù)組
bins = np.array([2, 4, 6])

# 將連續(xù)值轉(zhuǎn)換為離散值
discretized_values = np.digitize(values, bins)

print(discretized_values)

輸出結(jié)果為：

[0 1 2 2 3]

在這個例子中，連續(xù)值數(shù)組values中的元素被劃分為3個離散區(qū)間，分別對應(yīng)索引0、1和2。

0 踩