結果 : how to write if loop in python