shapely 笔记：STR TREE

数据结构笔记：R树-CSDN博客

1 基本介绍

• 使用Sort-Tile-Recursive (STR)  算法创建的仅查询的R-tree空间索引
• 该树索引每个几何图形的边界框。树在初始化时直接构建，且一旦创建后不能添加或移除节点
• 所有操作返回输入几何图形的索引
• 边界框限于二维并且是轴对齐的
  • 几何图形中存在的任何Z值在树内索引时都会被忽略。

2 创建R树

STRtree(geoms, 
        node_capacity)

3 属性

3.1 geometries

• 以用于构建树的顺序存储在树中的几何图形

from shapely import *
tree = STRtree([Point(i, i) for i in range(10)])
tree.geometries
'''
array([, , , ,
       , , , ,
       , ], dtype=object)
'''

4 方法

4.1 nearest

• 基于二维笛卡尔空间中的距离，返回树中每个输入几何图形的最近几何图形的索引
• 当输入几何图形与树几何图形相交时，此距离将为0
• 如果树中存在多个等距离或相交的几何图形，每个输入几何图形只返回一个结果，基于访问树几何图形的顺序；
  • 此顺序可能是不确定的

from shapely import *
tree = STRtree([Point(i, i) for i in range(10)])
 
 
'''
查询单个几何形状
'''
tree.nearest(Point(2.2,2.2))
#2
print(tree.geometries[2])
#POINT (2 2)
 
 
 
'''
查询多个几何形状
'''
tree.nearest([Point(2.2, 2.2), Point(4.4, 4.4)])
#array([2, 4], dtype=int64)
 
 
'''
如果存在等距的情况，只返回一个
'''
tree.nearest(Point(2.5,2.5))
#2

4.2 query

query(geometry, 
    predicate=None, 
    distance=None)

• 返回每个输入几何图形与树几何图形的所有组合的整数索引
  • 如果输入几何图形是标量，则返回形状为(n, )的数组，包含匹配树几何图形的索引
  • 如果输入几何图形是array_like，则返回形状为(2,n)的数组，其中子数组对应于输入几何图形的索引和与每个相关联的树几何图形的索引
• 如果提供了谓词，则首先基于输入几何图形的边界框查询树几何图形，然后进一步过滤满足当比较输入几何图形与树几何图形时的谓词的那些几何图形

4.2.1 举例

• 单个几何形状

from shapely import *
points = [Point(0, 0), Point(1, 1), Point(2,2), Point(3, 3)]
tree = STRtree(points)
 
tree.query(box(0, 0, 1, 1))
#array([0, 1], dtype=int64)
'''
表示查询几何体和创建R树的几何体中索引0，1的点有交集
'''

• 多个几何形状

tree.query([box(0, 0, 1, 1), box(2, 2, 3, 3)])
'''
array([[0, 0, 1, 1],
       [0, 1, 2, 3]], dtype=int64)
第一行表示 第几个查询几何体
第二行表示对应的查询几何体和创建R树的几何体的哪个索引有交集
'''
 
tree.query([box(0, 0, 1, 1), box(2, 2, 3, 3)]).T
'''
array([[0, 0],
       [0, 1],
       [1, 2],
       [1, 3]], dtype=int64)
这样每一行就是第几个查询集合体，对应创建R树的第几个几何体
'''

• 带predicate

tree.query(box(0, 0, 1, 1),
          predicate='contains')
#array([], dtype=int64)

4.3 query_nearest 最近几何图形的索引

基于二维笛卡尔空间中的距离，返回树中每个输入几何图形的最近几何图形的索引

【个人感觉，相当于nearest promax？】

query_nearest(
    geometry, 
    max_distance=None, 
    return_distance=False, 
    exclusive=False, 
    all_matches=True)

4.3.1 主要参数

4.3.2 举例

from shapely import *
points = [Point(0, 0), Point(1, 1), Point(2,2), Point(3, 3)]
tree = STRtree(points)
 

• 单个点

tree.query_nearest(Point(0.25, 0.25))
#array([0], dtype=int64)

• 多个点

返回的内容和query是一样的解释方法

tree.query_nearest([Point(2.5, 2.5), Point(1, 1)])
'''
array([[0, 0, 1],
       [2, 3, 1]], dtype=int64)
'''

•  all_matches

即使all_matches为False，返回的内容和nearest还是不一样；nearest返回的是一维数组，这边还是二维

tree.query_nearest([Point(2.5, 2.5), Point(1, 1)],
                  all_matches=False)
'''
array([[0, 1],
       [2, 1]], dtype=int64)
'''

• return_distance

tree.query_nearest(Point(0.25, 0.25),
                  return_distance=True)
#(array([0], dtype=int64), array([0.35355339]))