截断numpy.ndarray中的所有元素（浮点数）

Question

我想删除numpy数组中所有元素（float）的数字，以便在点后面有一个固定的位数，例如

[ 0.5029503212   0.50408515343  0.50531858897] -> [ 0.5029503   0.50408515  0.50531858]

我不想要任何四舍五入的浮点数。

谢谢

Answer 1

根据jonrsharpe的建议，allclose帮助我解决了这个问题。但首先我们需要将numpy ndarray转换为列表。这是我比较两个numpy ndarray数组的代码

  

中心 和 center_1

center_a = center.tolist()   # Convert numpy array into list
center_b = center_1.tolist() # Convert numpy array into list
np.allclose(center_b,center_a,rtol=1e-04, atol=1e-04)

当两个数组在指定的容差范围内在元素方面相等时，这将返回true。