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