我有一些涉及两个矩阵的计算,这两个矩阵都以numpy
数组表示。
计算后,我得到一个向量floats
表示在另一个numpy
数组中。
我想向上/向下舍入此结果向量中的值,例如,如果计算给出:
array([1.33333, 2.56, 9.99999, 16.0])
那么它应该四舍五入为:
array([1, 3, 10, 16])
最快的方法是什么?
NumPy 数组有一个round方法:
In [73]: x = np.array([1.33333, 2.56, 9.99999, 16.0])
In [74]: x.round()
Out[76]: array([ 1., 3., 10., 16.])