python - Numpy 行明智掩蔽

Question

我有一个 numpy 数组，它在某些位置有 NaN 值。

>>> d1
array([[  0.,   1.,   2.,  nan,   4.,   5.,  nan,   7.],
       [  8.,   9.,  10.,  nan,  12.,  13.,  nan,  15.],
       [ 16.,  17.,  18.,  nan,  20.,  21.,  nan,  23.],
       [ 24.,  25.,  26.,  nan,  28.,  29.,  30.,  31.],
       [ 32.,  33.,  34.,  35.,  36.,  37.,  38.,  39.],
       [ 40.,  41.,  42.,  43.,  44.,  45.,  46.,  47.],
       [ 48.,  49.,  50.,  51.,  52.,  53.,  54.,  55.],
       [ 56.,  57.,  58.,  59.,  60.,  61.,  62.,  63.]])

我想制作掩码，如果该行中存在 NaN 值，它将使整行成为 False 值。如下所示。

>>> mask
array([[False, False, False, False, False, False, False, False],
       [False, False, False, False, False, False, False, False],
       [False, False, False, False, False, False, False, False],
       [False, False, False, False, False, False, False, False],
       [ True,  True,  True,  True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True,  True,  True,  True]], dtype=bool)

我尝试像下面那样应用面膜，但它没有用。

mask = NaN not in d1[,:]

谁能帮我解决这个问题？

Answer 1

这里有几种方法-

np.repeat(~np.isnan(d1).any(1,keepdims=1),d1.shape[1],axis=1)
~np.isnan(d1).any(1,keepdims=1)*([True]*d1.shape[1])
np.tile(~np.isnan(d1).any(1,keepdims=1),d1.shape[1])
np.broadcast_to(~np.isnan(d1).any(1,keepdims=1), d1.shape)
np.broadcast_to(~np.isnan(d1).any(1), d1.shape).T