我有一个全球数字阵列数据这是一个200*200*3的三维数组,在三维空间中包含40000个点。在
我的目标是计算每个点到单位立方体四个角点的距离((0,0,0),(1,0,0),(0,1,0),(0,0,1)),这样我就可以确定哪个角点离它最近。在
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)
我写了这段代码并对其进行了测试,每次运行大约花费10毫秒。 我的问题是如何提高这段代码的性能,在不到1ms的时间内更好地运行。在
您可以使用^{} -
运行时测试
^{pr2}$时间安排-
相关问题 更多 >
编程相关推荐