結果 : how to write if condition in lambda python