結果 : how to restrict user input in python