Python Numpy Random Choice