結果 : for and if statement in one line python