atompy.Vector.keep_where#

Vector.keep_where(mask)[source]#

Keep every vector i where mask[i] == True

Parameters:
masknumpy.ndarray, shape (len(self),)

Array of booleans.

Returns:
Vector

Examples

>>> vec = Vector([[1, 2, 3], [4, 5, 6]])
[[1. 2. 3.]
 [4. 5. 6.]]
>>> vec.keep_where(vec.z == 3)
[[1. 2. 3.]]