Creating a NumPy masked array