python - 如何在numpy中进行条件行求和？

Question

我有一个 numpy 2d 数组，其中包括正条目和 -1 值，它们是缺失值。我想对这个矩阵的行求和，而不包括 -1 值。有没有办法在 numpy 中执行条件行求和？

Answer 1

方法 #1：乘以一个布尔数组，它的工作原理True是 1 和False0：

>>> a
array([[ 1,  3,  0,  4,  2],
       [ 2,  1,  4, -1,  1],
       [-1,  2,  4,  4,  4],
       [ 4,  4,  4,  4,  0],
       [ 0,  2,  4,  1,  2]])
>>> (a * (a != -1)).sum(axis=1)
array([10,  8, 14, 16,  9])

方法#2：使用相同的布尔数组（好吧，翻转 - 在我们想知道要保留哪些，现在要忽略哪些之前）作为 a 的正式掩码MaskedArray。

>>> am = np.ma.MaskedArray(a, mask=(a==-1))
>>> am
masked_array(data =
 [[1 3 0 4 2]
 [2 1 4 -- 1]
 [-- 2 4 4 4]
 [4 4 4 4 0]
 [0 2 4 1 2]],
             mask =
 [[False False False False False]
 [False False False  True False]
 [ True False False False False]
 [False False False False False]
 [False False False False False]],
       fill_value = 999999)

>>> am.sum(axis=1)
masked_array(data = [10 8 14 16 9],
             mask = [False False False False False],
       fill_value = 999999)

[不过说实话，我几乎从不使用MaskedArray.]