def take(a, indices, axis=0, fill_value=None):
"""
Selects the elements of `a` corresponding to `indices`.
"""
if _isPhysical(a):
taken = a.take(indices, axis=axis)
elif type(indices) is type(MA.array((0))):
## Replaces `MA.take`. `MA.take` does not always work when
## `indices` is a masked array.
##
taken = MA.take(a, MA.filled(indices, 0), axis=axis)
mask = MA.getmask(indices)
if mask is not MA.nomask:
mask = MA.getmaskarray(indices)
if taken.shape != mask.shape:
mask = MA.repeat(mask[NewAxis, ...], taken.shape[0], axis=0)
mask = MA.mask_or(MA.getmask(taken), mask)
if mask is not MA.nomask:
taken = MA.array(data=taken, mask=mask)
else:
if MA.getmask(taken) is MA.nomask and numpy_version == 'old':
# numpy 1.1 returns normal array when masked array is filled
taken = taken.filled()
elif type(a) in (type(array((0))), type(()), type([])):
taken = NUMERIX.take(a, indices, axis=axis)
elif type(a) is type(MA.array((0))):
taken = MA.take(a, indices, axis=axis)
else:
raise TypeError, 'cannot take from %s object: %s' % (type(a), `a`)
if fill_value is not None and type(taken) is type(MA.array((0))):
taken = taken.filled(fill_value=fill_value)
return taken
def take(a, indices, axis=0, fill_value=None):
"""
Selects the elements of `a` corresponding to `indices`.
"""
if _isPhysical(a):
taken = a.take(indices, axis=axis)
elif type(indices) is type(MA.array((0))):
## Replaces `MA.take`. `MA.take` does not always work when
## `indices` is a masked array.
##
taken = MA.take(a, MA.filled(indices, 0), axis=axis)
mask = MA.getmask(indices)
if mask is not MA.nomask:
mask = MA.getmaskarray(indices)
if taken.shape != mask.shape:
mask = MA.repeat(mask[NewAxis, ...], taken.shape[0], axis=0)
mask = MA.mask_or(MA.getmask(taken), mask)
if mask is not MA.nomask:
taken = MA.array(data=taken, mask=mask)
else:
if MA.getmask(taken) is MA.nomask and numpy_version == 'old':
# numpy 1.1 returns normal array when masked array is filled
taken = taken.filled()
elif type(a) in (type(array((0))), type(()), type([])):
taken = NUMERIX.take(a, indices, axis=axis)
elif type(a) is type(MA.array((0))):
taken = MA.take(a, indices, axis=axis)
else:
raise TypeError, 'cannot take from %s object: %s' % (type(a), ` a `)
if fill_value is not None and type(taken) is type(MA.array((0))):
taken = taken.filled(fill_value=fill_value)
return taken