Home > Software design >  Mask Indices in an array
Mask Indices in an array

Time:07-28

I have an array like:

data=np.array([1,2,3,5,8,7,2,1,3,5,1,2,20])

I would like to mask an array indices with a step of 3. For an example I able to mask where the value of an array equal to 3.

import numpy as np
import numpy.ma as ma
x = np.array([1,2,3,5,8,7,2,1,3,5,1,2,20])
mx=ma.masked_values(x,3)
output:
[1 2 -- 5 8 7 2 1 -- 5 1 2 20]

Requirement: I need to mask the every 3nd Indices in an array.(step of 3)

Required Output:

[1,2,3,--,8,7,--,1,3,--,1,2,--]

CodePudding user response:

Let us create a mask based on the indices of the array:

np.ma.masked_array(x, np.r_[1, 1:len(x)] % 3 == 0)

masked_array(data=[1, 2, 3, --, 8, 7, --, 1, 3, --, 1, 2, --],
             mask=[False, False, False,  True, False, False,  True, False,
                   False,  True, False, False,  True],
       fill_value=999999)

CodePudding user response:

try this:

mask = list(map(lambda i: i%3==2, np.arange(len(x))))
ma.masked_array(x, mask=mask)

Output:

masked_array(data=[1, 2, --, 5, 8, --, 2, 1, --, 5, 1, --, 20],
             mask=[False, False,  True, False, False,  True, False, False,
                    True, False, False,  True, False],
       fill_value=999999)
  • Related