查找给定数字在分区中的位置

Question

假设我有一个排序的整数数组说partition = [0, 3, 7, 12, 18, 23, 27]然后给出一个值value = 9我想返回我的价值所在的间隔。例如bounds = function(partition, value)print(bounds)>>>[7,12]是否有一个功能可以帮助我，还是我必须从头开始构建它？

Answer 1

试试numpy.searchsorted()。来自纪录片：

查找应插入元素以保持顺序的索引。

import numpy as np

partition = np.array( [0, 3, 7, 12, 18, 23, 27] )

value = 9

idx = np.searchsorted(partition,value)

bound = (partition[idx-1],partition[idx])

print(bound)

>>>>(7,12)

的优点searchsorted是它可以一次为您提供多个值的索引。

Answer 2

该对开模块是有效的这样做很好。它将返回上限的索引。

如果该值可能超出范围，您需要进行一些错误检查：

from bisect import bisect

partition = [0, 3, 7, 12, 18, 23, 27]

value = 9

top = bisect(partition, value)

print(partition[top-1], partition[top])

# 7 12

Answer 3

 def function(partition,value):

  for i in range(len(partition)):

  if partition[i]<value and partition[i+1]>value:

    print [partition[i],partition[i+1]]

 partition = [0, 3, 7, 12, 18, 23, 27,5,10]

 value=9

 function(partition,value)