結果 : how to take user input for array in python