Python 边界框内的numpy过滤器点
我有一个二维点列表Python 边界框内的numpy过滤器点,python,numpy,coordinates,bounding-box,Python,Numpy,Coordinates,Bounding Box,我有一个二维点列表[(x1,y1),(x2,y2),…,(xn,yn)]在numpy中,如何在角((bx1,by1),(bx2,by2))指定的边界框内获得点列表 如果这是C++,我会用它来过滤列表。< /P> 现在我正在处理NxN 2d numpy数组中的索引列表,所以我希望这应该是1-2行numpy代码。使用all、logical\u和以及Meaby的组合,您会发现这很有帮助:而且在我添加pyplot.show()之后,示例工作得非常好!谢谢@安德烈:谢谢你的编辑。我已经开始使用pyplot
[(x1,y1),(x2,y2),…,(xn,yn)]
在numpy中,如何在角((bx1,by1),(bx2,by2))指定的边界框内获得点列表
如果这是C++,我会用它来过滤列表。< /P>
现在我正在处理NxN 2d numpy数组中的索引列表,所以我希望这应该是1-2行numpy代码。使用all
、logical\u和
以及Meaby的组合,您会发现这很有帮助:而且在我添加pyplot.show()
之后,示例工作得非常好!谢谢@安德烈:谢谢你的编辑。我已经开始使用pyplot.ion()。
import random
import numpy as np
from matplotlib import pyplot
points = [(random.random(), random.random()) for i in range(100)]
bx1, bx2 = sorted([random.random(), random.random()])
by1, by2 = sorted([random.random(), random.random()])
pts = np.array(points)
ll = np.array([bx1, by1]) # lower-left
ur = np.array([bx2, by2]) # upper-right
inidx = np.all(np.logical_and(ll <= pts, pts <= ur), axis=1)
inbox = pts[inidx]
outbox = pts[np.logical_not(inidx)]
# this is just for drawing
rect = np.array([[bx1, by1], [bx1, by2], [bx2, by2], [bx2, by1], [bx1, by1]])
pyplot.plot(inbox[:, 0], inbox[:, 1], 'rx',
outbox[:, 0], outbox[:, 1], 'bo',
rect[:, 0], rect[:, 1], 'g-')
pyplot.show()