python - 将浮点值舍入到区间限制/网格

Question

我有一个(随机) float 数组。我想将每个值四舍五入到任意网格的限制。请参阅以下示例:

import numpy as np
np.random.seed(1)

# Setup
sample = np.random.normal(loc=20, scale=6, size=10)
intervals = [-np.inf, 10, 12, 15, 18, 21, 25, 30, np.inf]

# Round each interval up
for i in range(len(intervals) - 1):
    sample[np.logical_and(sample > intervals[i], sample <= intervals[i+1])] = intervals[i+1]

这导致:

[ 30.  18.  18.  15.  30.  10.  inf  18.  25.  21.]

如何避免 for 循环？我确信有一些方法可以使用 NumPy 的数组魔法，但我现在还没有看到。

Answer 1

如果intervals被排序，你可以使用np.searchsorted :

np.array(intervals)[np.searchsorted(intervals, sample)]
# array([ 30.,  18.,  18.,  15.,  30.,  10.,  inf,  18.,  25.,  21.])

---

searchsorted 返回元素所属区间的索引:

np.searchsorted(intervals, sample)
# array([7, 4, 4, 3, 7, 1, 8, 4, 6, 5])

默认 side='left' 返回该区间的最小索引，结果属于左开右闭场景。