how to flatten a numpy ndarray