結果 : how to write one line if statement in python