結果 : how to put user input into an array python