2
我有一个全球numpy.array 数据这是一个200 * 200 * 3的三维阵列在三维空间中包含40000点。 (0,0,0),(1,0,0),(0,1,0),(0,0,0,0),(1,0,0,0,0), 0,1)),所以我可以确定哪个角落离它最近。Python如何提高numpy数组的性能?
def dist(*point):
return np.linalg.norm(data - np.array(rgb), axis=2)
buffer = np.stack([dist(0, 0, 0), dist(1, 0, 0), dist(0, 1, 0), dist(0, 0, 1)]).argmin(axis=0)
我写了这段代码并进行了测试,每次运行约耗时10ms。 我的问题是如何提高这段代码的性能,更好地在不到1ms的时间内运行。
不立方体有超过4个角? –
@JohnZwinck只需要计算其中四个的距离。 – iouvxz