Python numpy real_if_close用法及代码示例

本文简要介绍 python 语言中 numpy.real_if_close 的用法。

用法:

numpy.real_if_close(a, tol=100)

如果输入是复数且所有虚部都接近于零，则返回实部。

“Close to zero” 定义为 tol *(a 类型的机器 epsilon)。

参数：

a： array_like

输入数组。

tol： 浮点数

数组中元素的复杂部分的机器 epsilon 容差。

返回：

out： ndarray

如果 a 是实数，则将 a 的类型用于输出。如果 a 具有复杂元素，则返回类型为 float。

注意：

机器 epsilon 因机器和数据类型而异，但大多数平台上的 Python 浮点数的机器 epsilon 等于 2.2204460492503131e-16。您可以使用“np.finfo(float).eps”打印出机器 epsilon 以获取浮点数。

例子：

>>> np.finfo(float).eps
2.2204460492503131e-16 # may vary

>>> np.real_if_close([2.1 + 4e-14j, 5.2 + 3e-15j], tol=1000)
array([2.1, 5.2])
>>> np.real_if_close([2.1 + 4e-13j, 5.2 + 3e-15j], tol=1000)
array([2.1+4.e-13j, 5.2 + 3e-15j])