python判断数字是否是inf或者nan

一共有两种方法

使用math模块

代码如下:

import math
a=float('inf')
b=float('nan')
c=1.0
print(math.isnan(a))
print(math.isinf(a))
print(math.isnan(b))
print(math.isinf(b))
print(math.isnan(c))
print(math.isinf(c))

使用numpy模块

代码如下:

import numpy as np
a=float('inf')
b=float('nan')
c=1.0
print(np.isnan(a))
print(np.isinf(a))
print(np.isnan(b))
print(np.isinf(b))
print(np.isnan(c))
print(np.isinf(c))