I have an array of coordinates as shown below:

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]])

As my ultimate goal, I need to choose the positions from the values_to_test that contain the specific coordinates of the following array:

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

Hence, I'm expecting a result like: [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

推荐答案

You could make use of numpy.all and numpy.any like so:

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

Python相关问答推荐

根据条件按以上 5 行过滤数据框

如何将 Torch 张量转换为 FastAPI 返回的图像?

将Pandas 数据框的两列与字符串列表进行比较

包含列表所有值的 Pandas DataFrame 行

使用 json_modify.py 更改 JSON 键名

查找掩码的周长坐标

Apache Airflow UI 显示 DAG 导入错误(IndexError: list index out of range)但 DAG 工作正常

Python 中的 Google BigQuery 查询在使用 result() 时有效,但在使用 to_dataframe() 时出现权限问题

函数将 DF 的所有列数据更改为相同的值,并忽略不同行的条件不同.并且 .Apply() 给出错误

列表理解是通过 map 和 lambda 函数实现的吗?

Pandas 数据框列值之间的比较

如何在视图中获取 Select 表单的值?

满足以下约束的纸浆程序 min(a,b) > min(x,y)

从 C++ 函数调用 python lambda 时处理 GIL

用 NaN Pandas 替换所有数值

处理丢失的 SSH 连接

如何使用 Pillow Python 获取带有换行符的文本大小

bert-base-uncased:TypeError:元组索引必须是整数或切片,而不是元组

Python - 获取调用我的行的源代码

连接几个Pandas 数据框