Boolean array with no all True values in one row

Question

I have numpy array:

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]]

Need in each row no all values True - so here can be only 0, 1 or 2 True because 3 'columns'.

Ugly solution is:

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]]

What is better and more generic solution?

mask[mask.all(1),-1] = 0? That would create a different one than the one posted with the ugly solution though. — Divakar
– Divakar, Commented Feb 7, 2017 at 14:30
@Divakar your shorter code version beat my answer by a good few seconds — EdChum
– EdChum, Commented Feb 7, 2017 at 14:32

EdChum · Accepted Answer · 2017-02-07 14:34:31Z

3

You could do:

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)

So just test row-wise using all and only set the 3rd col to False on this condition

Thanks to @Divakar, you could type less:

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)

So here the position arg axis is being set and 0 is being cast to boolean False otherwise is same

Some explanation, first use all with axis=1 to test row-wise if all are True.

Then we use that mask to mask the rows in the square brackets, the second arg -1 selects the last column and finally we assign the new desired value

edited Feb 7, 2017 at 14:34

answered Feb 7, 2017 at 14:30

EdChum

397k204 gold badges836 silver badges583 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

Divakar Over a year ago

I would encourage you to add that shorter one as both are basically the same thing, if you would like :)

EdChum Over a year ago

@Divakar sure will do

EdChum Over a year ago

@dabadaba I'm using the original array as a mask, we test if all values are True row-wise when we pass axis=1, the -1 sets the last column to the desired value

Paul Panzer · Accepted Answer · 2017-02-07 14:39:02Z

2

Here's a fair one (all legal triplets equally likely):

N = 10
bits = np.random.randint(7, size=(N,))
mask = (bits[:, None] & 2**np.arange(3)).astype(bool)

answered Feb 7, 2017 at 14:39

Paul Panzer

53.3k3 gold badges60 silver badges103 bronze badges

Collectives™ on Stack Overflow

Boolean array with no all True values in one row

2 Answers 2

3 Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

3 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related