清单理解条件中的`elif`

Question

我们可以elif在列表理解中使用吗？

范例：

l = [1, 2, 3, 4, 5]

for values in l:
    if values==1:
        print 'yes'
    elif values==2:
        print 'no'
    else:
        print 'idle'

我们可以elif采用与上述代码类似的方式将列表理解包括在内吗？

例如，答案如下：

['yes', 'no', 'idle', 'idle', 'idle']

到现在为止，我仅使用if和else理解列表。

Answer 1

Python的条件表达式正是针对这种用例而设计的：

>>> l = [1, 2, 3, 4, 5]
>>> ['yes' if v == 1 else 'no' if v == 2 else 'idle' for v in l]
['yes', 'no', 'idle', 'idle', 'idle']