将nan值转换为z

Question

我有一个二维核阵列。这个数组中的一些值是NaN。我想使用这个数组执行某些操作。例如，考虑数组：

[[   0.   43.   67.    0.   38.]
 [ 100.   86.   96.  100.   94.]
 [  76.   79.   83.   89.   56.]
 [  88.   NaN   67.   89.   81.]
 [  94.   79.   67.   89.   69.]
 [  88.   79.   58.   72.   63.]
 [  76.   79.   71.   67.   56.]
 [  71.   71.   NaN   56.  100.]]

我试着对每一行进行排序，一次一行，按照相反的顺序从行中获取最多3个值并取它们的平均值。我尝试的代码是：

# nparr is a 2D numpy array
for entry in nparr:
    sortedentry = sorted(entry, reverse=True)
    highest_3_values = sortedentry[:3]
    avg_highest_3 = float(sum(highest_3_values)) / 3

这对于包含NaN的行不起作用。我的问题是，是否有一种快速的方法可以将2D numpy数组中的所有NaN值转换为零，这样我就不会在排序和其他尝试做的事情上遇到问题。

Answer 1

这应该有效：

from numpy import *

a = array([[1, 2, 3], [0, 3, NaN]])
where_are_NaNs = isnan(a)
a[where_are_NaNs] = 0

在上述情况下，其中：

In [12]: where_are_NaNs
Out[12]: 
array([[False, False, False],
       [False, False,  True]], dtype=bool)

Answer 2

nan_to_num()怎么样？

Answer 3

其中A是二维数组：

import numpy as np
A[np.isnan(A)] = 0

函数isnan生成一个bool数组，指示NaN值的位置。布尔数组可用于索引相同形状的数组。把它想象成面具。