Input Array In Python Using Numpy