Numpy Array Multiple Choice