列表理解中的else語句錯誤

Question

我在玩列表理解，但是else語句出現語法錯誤

doctor = ['house', 'cuddy', 'chase', 'thirteen', 'wilson'] 
first = [doc[0] for doc in doctor if doc[0] == 'h' else doc[3]]

這是什么問題？

Answer 1

以下代碼

doc[0] for doc in doctor if doc[0] == 'h' else doc[3]

大致翻譯成

for doc in doctor
    if doc[0] == 'h'
        doc[0]
else doc[3]

因此else部分沒有doc定義。 正確的代碼是

first = [doc[0] if doc[0] == 'h' else doc[3] for doc in doctor]

其中

doc[0] if doc[0] == 'h' else doc[3]

是基於doc每個迭代值的三元條件。

Answer 2

你可以試試這個嗎？

doctor = ['house', 'cuddy', 'chase', 'thirteen', 'wilson'] 
first = [doc[0] if doc[0] == 'h' else doc[3] for doc in doctor]