问题描述
请考虑以下问题:给定一组 n 个间隔和一组 m 个浮点数,为每个浮点数确定包含浮点数的时间间隔。
已通过构造interval tree(或称为范围树或分段树)解决了此问题。已经针对一维情况完成了实现,例如python的intervaltree package。通常,这些实现会考虑一个或几个浮点数,即上面的小“ m”。
在我的问题设置中,n和m都是非常大的数字(来自于解决图像处理问题)。此外,我需要考虑N维间隔(当N = 3时称为长方体,因为我使用有限元方法对人脑进行建模)。我在python中实现了一个简单的N维间隔树,但是它循环运行,一次只能取一个浮点数。任何人都可以帮助提高实施效率吗?您可以自由更改数据结构。
import sys
import time
import numpy as np
# find the index of a satisfying x > a in one dimension
def find_index_smaller(a,x):
idx = np.argsort(a)
ss = np.searchsorted(a,x,sorter=idx)
res = idx[0:ss]
return res
# find the index of a satisfying x < a in one dimension
def find_index_larger(a,x):
return find_index_smaller(-a,-x)
# find the index of a satisfing amin < x < amax in one dimension
def find_intv_at(amin,amax,x):
idx = find_index_smaller(amin,x)
idx2 = find_index_larger(amax[idx],x)
res = idx[idx2]
return res
# find the index of a satisfying amin < x < amax in N dimensions
def find_intv_at_nd(amin,x):
dim = amin.shape[0]
res = np.arange(amin.shape[-1])
for i in range(dim):
idx = find_intv_at(amin[i,res],amax[i,x[i])
res = res[idx]
return res
我还有两个用于健全性检查和性能测试的测试示例:
def demo1():
print ("By default,we do a correctness test")
n_intv = 2
n_point = 2
# generate the test data
point = np.random.rand(3,n_point)
intv_min = np.random.rand(3,n_intv)
intv_max = intv_min + np.random.rand(3,n_intv)*8
print ("point ")
print (point)
print ("intv_min")
print (intv_min)
print ("intv_max")
print (intv_max)
print ("===Indexes of intervals that contain the point===")
for i in range(n_point):
print (find_intv_at_nd(intv_min,intv_max,point[:,i]))
def demo2():
print ("Performance:")
n_points=100
n_intv = 1000000
# generate the test data
points = np.random.rand(n_points,3)*512
intv_min = np.random.rand(3,n_intv)*512
intv_max = intv_min + np.random.rand(3,n_intv)*8
print ("point.shape = "+str(points.shape))
print ("intv_min.shape = "+str(intv_min.shape))
print ("intv_max.shape = "+str(intv_max.shape))
starttime = time.time()
for point in points:
tmp = find_intv_at_nd(intv_min,point)
print("it took this long to run {} points,with {} interva: {}".format(n_points,n_intv,time.time()-starttime))
我的想法是:
任何贡献将不胜感激。
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)