我想计算列表的非重叠子序列中的元素数量,例如,如下所示:
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
如果将原始数组分解为子数组,则可以轻松解决此问题,但是在包含数千个数字的数组中,此类方法的计算量很大。 在上面的示例中,我将间隔显示为输出,这只是为了解释我的问题。我只想要每个间隔的计数。 我可以帮忙吗?
谢谢。
答案 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