Numpy Random Choice 2d