結果 : python create a column based on condition