给定x-y坐标的两个点,我指定了一条线。现在我想在2-d numpy数组中区分左边的点和右边的点。
以下代码可以解决问题 - 但是双循环会伤害我的宗教感情(朝着numpy)。必须有一个更聪明的方法吗?
null
答案 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)