How To Give Input As Array In Python