Numpy 位置和向量数组 - 实现碰撞检测

Question

我在 PyOpenGL 中使用点精灵与 numpy 和 glDrawArrays。所以我有两个数组，一个用于点，一个用于向量。

        r = lambda: random.random()        
        self.pts = numpy.zeros((2000,2), dtype=numpy.uint16)
        for pt in self.pts:
            pt[0] = 300*r()
            pt[1] = 200*r()

        self.vectors = numpy.zeros((2000,2), dtype=numpy.uint16)
        for vec in self.vectors:
            vec[0] = 3*r()
            vec[1] = 3*r()

现在我需要根据与屏幕边框的碰撞来更新位置和向量数组。例如，

if pt[0][0]-width < 0: pt[0][0] = width; vec[0][0] *= -1

最后我必须在屏幕上有一个 2000x2 的 pt 数组来提供给 opengl。

编辑 - 将点保留在边界框内的当前解决方案（0,0,宽度,高度）

points [:,0][points[:,0] > width] = width
vectors[:,0][points[:,0] > width] *= -1
points [:,0][points[:,0] < 0] = 0
vectors[:,0][points[:,0] < 0] *= -1
points [:,1][points[:,1] > height] = height
vectors[:,1][points[:,1] > height] *= -1
points [:,1][points[:,1] < 0] = 0
vectors[:,1][points[:,1] < 0] *= -1

Answer 1

如果我理解你的问题，你想做这样的事情：
（试图在这里说明几种不同的方法）

import numpy as np

numpoints, numdimensions = 2000, 2

# Generate random points
points = np.random.random((numpoints, numdimensions))
points[:,0] *= 300
points[:,1] *= 200
points = points.astype(np.int16)

# Generate random vectors
vectors = np.random.randint(-3, 3, size=(numpoints, numdimensions)).astype(np.int16)

# Update points and vectors where point x-coords are > width 
points[:,0][points[:,0] > width] = width
vectors[:,0][points[:,0] > width] = -1