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