Python中的np.round()函数详解

1. 概述

在Python中，np.round()是NumPy库中的一个函数，用于对数组或单个数值进行四舍五入。该函数返回一个与输入类型相同的数组或数值，并可以通过可选的参数来指定保留的小数位数。

在本文中，我们将详细介绍np.round()函数的使用方法、参数和示例等内容。

2. np.round()函数的语法

np.round(a, decimals=0, out=None)

参数说明

a：待进行四舍五入操作的数组或单个数值。
decimals：可选参数，指定保留的小数位数，默认为0。
out：可选参数，无需手动指定。

返回值

返回与输入类型相同的数组或数值。

3. np.round()函数的示例

3.1 对单个数值进行四舍五入

import numpy as np

# 对数值进行四舍五入
number = 3.14159
rounded_number = np.round(number)
print(rounded_number)  # 输出: 3.0

在上述示例中，我们首先导入了numpy库，并定义了一个名为number的变量，其值为3.14159。然后，我们使用np.round()函数对number进行四舍五入操作，并将结果赋值给rounded_number变量。最后，我们使用print()函数输出rounded_number的值，结果为3.0。

3.2 对数组进行四舍五入

import numpy as np

# 对数组进行四舍五入
arr = np.array([1.1, 2.6, 3.3, 4.8, 5.5])
rounded_arr = np.round(arr)
print(rounded_arr)  # 输出: [1. 3. 3. 5. 6.]

在上述示例中，我们定义了一个名为arr的数组，其包含了几个浮点数元素。然后，我们使用np.round()函数对arr进行四舍五入操作，并将结果赋值给rounded_arr变量。最后，我们使用print()函数输出rounded_arr的值，结果为[1. 3. 3. 5. 6.]。

3.3 指定保留的小数位数

import numpy as np

# 指定保留2位小数
arr = np.array([1.123, 2.456, 3.789])
rounded_arr = np.round(arr, decimals=2)
print(rounded_arr)  # 输出: [1.12 2.46 3.79]

在上述示例中，我们使用np.round()函数对数组arr进行四舍五入操作，并通过decimals参数指定保留两位小数。最后，我们使用print()函数输出rounded_arr的值，结果为[1.12 2.46 3.79]。

3.4 将结果赋值给指定的输出数组

import numpy as np

# 将结果赋值给指定的输出数组
arr = np.array([1.1, 2.6, 3.3, 4.8, 5.5])
out_arr = np.zeros_like(arr)
np.round(arr, out=out_arr)
print(out_arr)  # 输出: [1. 3. 3. 5. 6.]

在上述示例中，我们首先创建了一个与arr具有相同形状的全零数组out_arr。然后，我们使用np.round()函数对arr进行四舍五入操作，并将结果赋值给out_arr数组。最后，我们使用print()函数输出out_arr的值，结果与之前的示例相同。