np.linalg.norm 以及如何处理机器 epsilon

Question

我有这个相当简单的问题。 我想使用以下代码计算 numpy 的欧几里得距离：

a= np.array([1,2,3])
b= np.array([2,3,4])

print((np.linalg.norm(a-b))**2)

这产生2.9999999999999996 ，但是，答案应该是3 。 我如何实现这一目标？

谢谢

Answer 1

通常，您可以使用np.isclose来比较浮点值。

Answer 2

您可以使用 round() function

a= np.array([1,2,3])
b= np.array([2,3,4])

print((np.linalg.norm(round(a-b))**2))