How To Input An Array In Python