python上下对齐,Python：上下对齐布尔型numpy数组的索引

How can I up and down justify a numpy bool array. By Justify, I mean take the True values, and move them so they are either the first values at the top (if they are up justified) or the first values at the bottom. (if they are down justified)

[[False True True True True True]

[False False True True False True]

[False True False True False False

[True True True True False True]]

So If I down justify the True values of the bool array shown above, it would look like:

[[False False False True False False]

[False True True True False True]

[False True True True False True]

[True True True True True True]]

解决方案

Simply sort it along each column, which pushes down the True values, while brings up the False ones for down-justified version. For up-justified one, do a flipping on the sorted version.

Sample run to showcase the implementation -

In [216]: mask

Out[216]:

array([[False, True, True, True, True, True],

[False, False, True, True, False, True],

[False, True, False, True, False, False],

[ True, True, True, True, False, True]], dtype=bool)

In [217]: np.sort(mask,0) # Down justified

Out[217]:

array([[False, False, False, True, False, False],

[False, True, True, True, False, True],

[False, True, True, True, False, True],

[ True, True, True, True, True, True]], dtype=bool)

In [218]: np.sort(mask,0)[::-1] # Up justified

Out[218]:

array([[ True, True, True, True, True, True],

[False, True, True, True, False, True],

[False, True, True, True, False, True],

[False, False, False, True, False, False]], dtype=bool)