split a multidimensional numpy array using a condition,"good_data = np.array([x for x in data[(0), :] if x == 1.0])"