結果 : how to put an input in an if statement python