結果 : python array comprehension if else