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

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 个位置中 Select 位置:

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

因此,我期望得到如下结果:[True,False,False,False,True,False,False,True,False]

I know if this was a 1D array I could use np.where but I cannot think about a way to handle this problem.
Appreciate your help

推荐答案

你可以像这样使用numpy.allnumpy.any:

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

Python相关问答推荐

如何在msgraph.GraphServiceClient上进行身份验证?

try 与gemini-pro进行多轮聊天时出错

如何检测背景有噪的图像中的正方形

比较2 PD.数组的令人惊讶的结果

如何列举Pandigital Prime Set

如何使用根据其他值相似的列从列表中获取的中间值填充空NaN数据

优化器的运行顺序影响PyTorch中的预测

使用Python更新字典中的值

有没有一种ONE—LINER的方法给一个框架的每一行一个由整数和字符串组成的唯一id?

在代码执行后关闭ChromeDriver窗口

如果包含特定值,则筛选Groupby

Python pint将1/华氏度转换为1/摄氏度°°

如果有2个或3个,则从pandas列中删除空格

在我融化极点数据帧之后,我如何在不添加索引的情况下将其旋转回其原始形式?

提取最内层嵌套链接

Django在一个不是ForeignKey的字段上加入'

有了Gekko,可以创建子模型或将模型合并在一起吗?

EST格式的Azure数据库笔记本中的当前时间戳

如何在PYTHON中向单元测试S Side_Effect发送额外参数?

牛郎星直方图中分类列的设置顺序