Numpy Choose From Array To Number