結果 : how to code an if then statement in python