NumPy可以使用numpy.digitize
函数将连续值转换为离散值。numpy.digitize
函数接受两个参数,第一个参数为要转换的连续值,第二个参数为用来划分连续值的边界数组。函数返回一个整数数组,表示每个连续值所属的离散区间索引。
以下是一个示例代码,将连续值转换为离散值:
import numpy as np
# 创建连续值数组
values = np.array([1.5, 2.5, 3.5, 4.5, 5.5])
# 创建边界数组
bins = np.array([2, 4, 6])
# 将连续值转换为离散值
discretized_values = np.digitize(values, bins)
print(discretized_values)
输出结果为:
[0 1 2 2 3]
在这个例子中,连续值数组values
中的元素被划分为3个离散区间,分别对应索引0、1和2。