Python中的 numpy.isinf()

在使用Python进行数值计算时，经常会遇到一些特殊的数值，如无穷大（infinity）和非数字（NaN）。numpy.isinf()是用于判断一个数是否为无穷大的函数。

用法

numpy.isinf()的用法如下：

numpy.isinf(x, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])

参数说明：

• x：需要判断是否为无穷大的数。

• out：输出结果的数组。默认为None，表示自动生成一个与x有相同形状和数据类型的数组。

• where：True或者数组。表示如果要判断的元素为True，则将相应位置的值替换为true_value。如果要判断的元素为False，则将相应位置的值替换为false_value。默认为True。

返回值

numpy.isinf()函数的返回值是一个布尔类型的数组，数组元素的值表示相应位置的数是否为无穷大。如果是则返回True，否则返回False。

示例

import numpy as np

a = np.array([1, 2, np.inf, -np.inf, np.nan, 0])
print(np.isinf(a))

运行结果为：

[False False  True  True False False]

可以看到，在数组a中，第3个和第4个元素为无穷大，所以对应的结果为True；其他元素不是无穷大，对应的结果为False。

注意事项

• numpy.isinf()函数对于非数字（NaN）并不返回True或False，而是返回False；

• 如果要判断一个复数是否为无穷大，则只能使用numpy.isinf()的where参数来实现。例如：

  import numpy as np
  
  x = np.array([1+1j, 2-2j, np.inf, -np.inf, np.nan])
  y = np.zeros_like(x, dtype=bool)
  np.isinf(x.real, out=y.real)
  np.isinf(x.imag, out=y.imag)
  print(y)
  

  运行结果为：

  [[False False]
   [False False]
   [ True False]
   [ True False]
   [False False]]
  

  可以看到，y数组元素的值表示相应位置的数是否为无穷大或非数字。其中，第3和第4行表示一个复数的实部或虚部为无穷大；第5行表示一个复数为非数字。