How To Get Input Of Array In Python