python - 如何在numpy中求和NaN？

Question

我必须对 np.average 获得的两个值求和作为

for i in x :
    a1 = np.average(function1(i))
    a2 = np.average(function2(i))
    plt.plot(i, a1+a2, 'o')

但是 np.average可能会返回 NaN .那么，只有两个 a1 的点和 a2可用将被计算。

如何使用零代替 NaN对所有积分求和？

我试图在 numpy 中找到一个函数来这样做，但是 numpy.nan_to_num用于数组。

Answer 1

您可以像这样使用 numpy:

import numpy as np

a = [1, 2, np.nan]
a_sum = np.nansum(a)
a_mean = np.nanmean(a)

print('a = ', a) # [1, 2, nan]
print("a_sum = {}".format(a_sum)) # 3.0
print("a_mean = {}".format(a_mean)) # 1.5