给出两点逐行分离

时间:2017-12-08 13:55:10

标签: python numpy geometry line

给定x-y坐标的两个点,我指定了一条线。现在我想在2-d numpy数组中区分左边的点和右边的点。

以下代码可以解决问题 - 但是双循环会伤害我的宗教感情(朝着numpy)。必须有一个更聪明的方法吗?

null

1 个答案:

答案 0 :(得分:2)

可以使用ogrid

对其进行矢量化
x, y = np.ogrid[:100, :100]
start = (b_x - a_x)*(y - a_y) - (x - a_x)*(b_y - a_y) <= 0

这会给你一个布尔掩码。如果你想要一些其他的dtype

start = start.astype(np.int)