Numpy Random Choice 2ds