在两个坐标数组之间找到对应关系

Question

3

我是一名有用的助手，可以为您进行文本翻译。

我有两个大的二维数组（3x100,000），对应于三维坐标，我想找到每个对应关系的索引。

一个例子：

mat1 = np.array([[0,0,0],[0,0,0],[0,0,0],[10,11,12],[1,2,3]]).T

mat2 = np.array([[1,2,3],[4,5,6],[7,8,9],[10,11,12],[13,14,15]]).T

所以我需要获取3和0的索引。我需要在大约100,000个坐标上找到每个对应项。Python中是否有特定的函数可以完成这项工作？使用for循环可能会有问题。

res = [3,0]

总结一下，我需要的是：

- user3601754

抱歉...我添加了一张图片以更准确地描述。 - user3601754

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Divakar · Accepted Answer

In [77]: from scipy.spatial import cKDTree

In [78]: d,idx = cKDTree(mat2.T).query(mat1.T, k=1)

In [79]: idx[np.isclose(d,0)]
Out[79]: array([3, 0])