如何使用 NumPy 将数组元素舍入到给定的小数位数？

介绍

NumPy 是一个用于科学计算的 Python 库，其中包含了很多数组操作的函数。其中一个非常有用的函数是 around()，可以将数组元素舍入到给定的小数位数。

方法

around() 函数的基本语法如下：

numpy.around(a, decimals=0)

其中，参数 a 代表需要进行舍入的数组，参数 decimals 代表需要保留的小数位数。默认情况下，decimals 的值为 0。

例如，我们有一个包含浮点数的数组：

import numpy as np

a = np.array([1.23456789, 2.34567890, 3.45678901])

我们想将这些浮点数保留到小数点后两位：

np.around(a, decimals=2)

运行结果为：

array([1.23, 2.35, 3.46])

即将数组中的所有浮点数舍入到小数点后两位，并返回一个新的数组。

总结

使用 NumPy 的 around() 函数可以轻松地将数组元素进行舍入操作。它对于需要精确控制浮点数精度的科学计算非常有用。