KDTree.query

`KDTree.query`是一个函数，用于在给定的KDTree中查询每个查询点的k个最近邻居。它的语法如下：

dists, indices = kd_tree.query(x, k=1, eps=0, p=2, distance_upper_bound=np.inf)

其中，参数含义如下：

- `kd_tree`: 要查询的KDTree对象。
- `x`: 用于查询最近邻居的点集，可以是单个点或多个点，形状为(N, M)，其中N是点的数量，M是每个点的维度。
- `k`: 每个查询点要返回的最近邻居数量。默认为1。
- `eps`: 查询半径，表示与每个查询点的距离小于等于`eps`的点都将被视为最近邻居。默认为0，表示只查询k个最近邻居。
- `p`: 距离度量的指数，默认为2，表示使用欧几里得距离。
- `distance_upper_bound`: 距离的上界，如果查询点与任何点之间的距离大于此值，则不返回该点。默认为`numpy.inf`，表示不设置上界。

该函数将返回两个数组：

- `dists`: 形状为(N, k)，其中第i行包含查询点i的k个最近邻居与其之间的距离，按升序排列。
- `indices`: 形状为(N, k)，其中第i行包含查询点i的k个最近邻居的索引，按升序排列。