python - 一行中没有所有 True 值的 boolean 数组

Question

我有 numpy 数组:

np.random.seed(100)
mask = np.random.choice([True, False], size=(10,3))
print (mask)
[[ True  True False]
 [False False False]
 [ True  True  True] <- problem - all values True
 [ True  True False]
 [ True  True  True] <- problem - all values True
 [ True False  True]
 [ True False  True]
 [False  True  True]
 [ True False False]
 [False  True  True]]

每一行都不需要所有值 True - 所以这里只能是 0、1 或 2 True 因为 3 个“列”。

丑陋的解决方案是:

mask[:, -1] = False
print (mask)
[[ True  True False]
 [False False False]
 [ True  True False]
 [ True  True False]
 [ True  True False]
 [ True False False]
 [ True False False]
 [False  True False]
 [ True False False]
 [False  True False]]

什么是更好、更通用的解决方案？

Answer 1

你可以这样做:

In [109]:
mask[mask.all(axis=1),-1] = False
mask

Out[109]:
array([[ True,  True, False],
       [False, False, False],
       [ True,  True, False],
       [ True,  True, False],
       [ True,  True, False],
       [ True, False,  True],
       [ True, False,  True],
       [False,  True,  True],
       [ True, False, False],
       [False,  True,  True]], dtype=bool)

因此，只需使用 all 逐行测试，并在此条件下仅将第 3 列设置为 False

感谢@Divakar，你可以少输入:

In [110]:
mask[mask.all(1),2] = 0
mask

Out[110]:
array([[ True,  True, False],
       [False, False, False],
       [ True,  True, False],
       [ True,  True, False],
       [ True,  True, False],
       [ True, False,  True],
       [ True, False,  True],
       [False,  True,  True],
       [ True, False, False],
       [False,  True,  True]], dtype=bool)

所以这里位置 arg axis 被设置并且 0 被转换为 boolean 值 False 否则是相同的

一些解释，首先使用 all 和 axis=1 来测试 row-wise 是否都是 True。

然后我们使用该掩码来屏蔽方括号中的行，第二个参数 -1 选择最后一列，最后我们分配新的所需值