結果 : how to write one line if statement python