python 去除nan

你可以使用 numpy 库中的 isnan 函数来判断一个数是否为 NaN，然后使用 numpy 中的 replace 函数将其替换为其他值或者去除。例如：

import numpy as np

a = np.array([1, 2, np.nan, 4, np.nan])
a = np.where(np.isnan(a), 0, a)  # 将 NaN 替换为 0
print(a)  # [1. 2. 0. 4. 0.]

如果你想删除数组中的 NaN，可以使用 numpy 中的 isnan 函数和 logical_not 函数来实现。例如：

import numpy as np

a = np.array([1, 2, np.nan, 4, np.nan])
a = a[~np.isnan(a)]  # 删除 NaN
print(a)  # [1. 2. 4.]