Remove Columns Where Every Value Is Masked

I want to remove columns from a masked array where every value in the column is masked. So in the following example: >>> import numpy as np >>> test = np.array([[

Solution 1:

You can use fancy indexing:

test[:, ~np.all(test.mask, axis=0)]
#masked_array(data =
# [[1 --]
# [-- 3]
# [1 4]],
#             mask =
# [[False  True]
# [ True False]
# [False False]],
#       fill_value = 0)

Solution 2:

test[:, ~np.all(test == 0, axis=0)]

Solution 3:

In [13]: test
Out[13]:
masked_array(data =
 [[1 -- --]
 [-- 3 --]
 [1 4 --]],
             mask =
 [[False  True  True]
 [ True False  True]
 [False False  True]],
       fill_value = 0)


In [14]: test[:, :2]
Out[14]:
masked_array(data =
 [[1 --]
 [-- 3]
 [1 4]],
             mask =
 [[False  True]
 [ True False]
 [False False]],
       fill_value = 0)