python 将ndarry中的浮点数进行四舍五入取整

你可以使用 numpy 库中的 round() 函数来对 ndarray 中的浮点数进行四舍五入取整。示例代码如下：

import numpy as np

a = np.array([1.2, 2.7, 3.5, 4.1, 5.9])
b = np.round(a)  # 对 a 中的浮点数进行四舍五入取整
print(b)  # 输出 [1. 3. 4. 4. 6.]

其中，np.round() 函数默认对浮点数进行四舍五入取整，你也可以通过修改参数 decimals 来控制保留的小数位数，例如：

import numpy as np

a = np.array([1.2345, 2.6789, 3.5432, 4.1234, 5.9876])
b = np.round(a, decimals=2)  # 对 a 中的浮点数进行四舍五入取两位小数
print(b)  # 输出 [1.23 2.68 3.54 4.12 5.99]