计算数组的非重叠子间隔中的元素数

时间:2018-06-26 12:16:51

标签: python arrays python-3.x counting

我想计算列表的非重叠子序列中的元素数量,例如,如下所示:

lst = [0.2,1.3,2.2,3.5]
L = 0.5
for x in lst:
lb = x   # lowebound
ub = x+L   # upperbound
print(lb,ub)

输出为

  0.2 0.7  count=1
  1.3 1.8  count=1
  2.2 2.7  count=1
  3.5 4.0  count=1

我们看到,这些间隔是不重叠。同样,在每个间隔[lb,ub]中,我们只有一个元素属于lst,因此每个间隔count = 1。 但是,情况如下所示

 nlst = [0.2,0.6,2.2,3.8]

使用上面的代码将创建我们不想要的重叠间隔。例如使用上面的代码,我们得到

  0.2 0.7
  0.6 1.1
  2.2 2.7
  3.8 4.3

现在,由于在第一个[0.2,0.7]间隔中包含0.6,所以存在我们不希望出现的重叠。所需的输出是

  0.2 0.7   count = 2
  2.2 2.7   count = 1
  3.8 4.3   count = 1

如果将原始数组分解为子数组,则可以轻松解决此问题,但是在包含数千个数字的数组中,此类方法的计算量很大。 在上面的示例中,我将间隔显示为输出,这只是为了解释我的问题。我只想要每个间隔的计数。 我可以帮忙吗?

谢谢。

2 个答案:

答案 0 :(得分:1)

怎么样?

lst = [0.2, 0.6, 2.2, 3.8]
L = 0.5

count = None
counts = []
append = counts.append
ub = lst[0]

for x in lst:
    if x < ub:
        count += 1
    else:
        append(count)
        count = 1
        ub = x + L

counts = counts[1:] + [count]

print(counts)

更新:如果您也想收集间隔:

lst = [0.2, 0.6, 2.2, 3.8]
L = 0.5

ival = count = None
ivals, counts = [], []
ival = (None, lst[0])

for x in lst:
    if x < ival[1]:
        count += 1
    else:
        ivals.append(ival)
        counts.append(count)
        ival = x, x + L
        count = 1

ivals = ivals[1:] + [ival]
counts = counts[1:] + [count]

print(ivals)
print(counts)

或者,您可以通过类似的方式创建一个包含三元组(下限,上限,计数)的列表。

答案 1 :(得分:1)

这将起作用

list = [0.2,0.3,0.4,2.2,2.3,3.5,3.8,3.9,7.1]
interval = 0.5
list.sort()
answer = []
def calculate (list):
  count = 1
  check = list[0]
  del list[0] 
  list_copy = [i for i in list]
  for index, item in enumerate(list_copy):
    if check + interval > list_copy[index] :
      count += 1
      list.remove(item)
  answer.append(count)
  if (len(list)):
    calculate(list)
  else:
    return 
calculate(list);
print answer