python - 从坐标数组中查找特定坐标的位置

Question

我有一个坐标数组，如下所示:

import numpy as np
values_to_test = np.array([[1 , 0],[2 , 0],[2 , 0.5],[1 , 0.5],[0 , 0],[2 , 1], [0 , 0.5],[1 , 1],[0 , 1]])

作为我的最终目标，我需要从 values_to_test 中选择包含以下数组的特定坐标的位置:

specific_values=np.array([[1,0],[0,0],[1,1]])

因此，我期待这样的结果:[True,False,False,False,True,False,False,True,False]

我知道如果这是一个 1D 数组我可以使用 np.where 但我想不出解决这个问题的方法。
感谢您的帮助

Answer 1

您可以使用 numpy.all和 numpy.any像这样:

>>> (values_to_test[:, None] == specific_values).all(axis=-1).any(axis=-1)
array([ True, False, False, False,  True, False, False,  True, False])