結果 : how to store user input in array python