Question

因此，我有两个形状（p，2）的a和形状（q，2）的b的numpy数组。我用scipy创建一个KDTree

c=sp.KDTree(a)

我有一个上限，ub=1.0/12.0

print c.query(b,distance_upper_bound=ub)

我得到([inf,inf,inf...],[len(a),len(a),len(a)...]) 这意味着a和b没有共同的元素。为了确认这一点，我做了一个for循环

for n in range(len(c.data)):#the result is the same with len(a)
  for m in range(len(b)):
    if (c.data[m][0]-b[n][0])**2.0+(c.data[m][1]-b[n][1])**2.0<ub**2.0:
      print c.data[m],b[n],n,m

会打印出许多n,m对。
我已经将a和b数组设为通用数组，因为它发生在不同的数组上。
对我而言，最神秘的是，在同一代码中，我使用了query并获得了不错的结果。
那么，您能指出我正确的方向吗？

编辑：根据要求提供示例数据。这些是for周期打印的一些数据点。整个阵列真的很长。

a=np.array([[276.95368542701721, 330.18454774620238], 
[276.95368542701721, 330.18454774620238], 
[283.19923346114763, 337.60512985013065], 
[270.32420807690886, 331.46587512659323], 
[271.32525610216351, 333.51103014735435], 
[271.9742523815284, 330.26777673087207], 
[268.89584462538102, 331.5474437183201], 
[278.6808380388178, 331.92691700030088], 
[271.36541507290735, 332.74113908742231]])

b=np.array([[ 276.956177, 330.183134],
[ 276.956177, 330.183134],
[ 283.264282, 337.592966],
[ 270.319366, 331.538029],
[ 271.351807, 333.578056],
[ 272.019257, 330.268417],
[ 268.913958, 331.523153],
[ 278.681976, 331.927687],
[ 271.303917, 332.706767]])

Scipy的KDTree查询找不到任何内容

0 个答案: