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