整合 numba 和 Rtree.nearest()

问题描述

我一直在拼命地希望将 @numba.njit(parallel=True) 应用于 rtree.nearest() 函数，但我收到了“没有 pyobject”的错误声明。

由于 Rtree 是一个 ctypes 包装函数，我如何应用 ctypes 库，然后在 Rtree.nearest 上使用 numba？

伪代码如下：

Initialise variable coordinates_tree
Initialise variable MAX
Initialise variable 2d_coordinates_array
Initialise variable EMPTY euclidean_distance_array

Function start

    Start for loop (MAX):
        
        Change 2d_coordinates to tuple

        Use rtree.nearest to find the nearest coordinate in tree (in form of object id)

        Convert to list to retrieve the nearest coordinate

        Convert to nearest coordinate numpy array   

        Calculate Euclidean distance between 2d_coordinate and nearest_coordinate

        Store the Euclidean distance in euclidean_distance_array

    Return euclidean_distance_array_array

以下是我一直在调查的文档：https://numba.readthedocs.io/en/stable/user/cfunc.html

本质上，我是否必须确保输入和输出是 ctypes？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）