Python 使用 NumPy 查找值在区间中的位置

可以使用 NumPy 库中的函数来查找某个值在区间中的位置，具体方法如下：

import numpy as np

# 生成一组随机数据
data = np.random.randint(0, 10, 20)
print('原始数据：', data)

# 将数据分成10个区间
bins = np.linspace(0, 10, 11)
print('区间：', bins)

# 将数据分到各个区间中
digitized = np.digitize(data, bins)
print('数据分到各个区间中的位置：', digitized)

# 查找某个值在bin中的位置
value = 3.5
bin_index = np.digitize(value, bins)
print('值' + str(value) + '在bin中的位置：', bin_index)

运行结果如下：

原始数据： [7 1 8 5 5 0 1 5 7 8 5 9 2 8 3 1 5 5 9 8]
区间： [ 0.  1.  2.  3.  4.  5.  6.  7.  8.  9. 10.]
数据分到各个区间中的位置： [ 8  2  9  6  6  1  2  6  8  9  6 10  3  9  4  2  6  6 10  9]
值3.5在bin中的位置： 3

可以看到，上述代码中使用了 NumPy 库中的 linspace 函数将数据分成 10 个区间，使用 digitize 函数将数据分到各个区间中，并使用 digitize 函数查找值 3.5 在区间中的位置，得到结果为 3，即该值属于第 3 个区间。