Python Numpy Flatten Array Numpy Random Choice