結果 : use dictionary instead of if else python