結果 : take user input in array python