python - 使用 if 和 break 创建 Python 列表理解

Question

是否可以将此代码转换为列表理解？

for i in userInput:
    if i in wordsTask:
        a = i
        break

我知道如何转换其中的一部分:

[i for i in userInput if i in wordsTask]

但我不知道如何添加中断，文档也没有太大帮助。

如有任何帮助，我们将不胜感激。

Answer 1

a = next(i for i in userInput if i in wordsTask)

稍微分解一下:

[i for i in userInput if i in wordsTask]

将生成一个列表。您想要的是列表中的第一项。一种方法是使用下一个函数:

next([i for i in userInput if i in wordsTask])

Next 从迭代器返回下一个项目。对于像列表这样的可迭代对象，它最终会获取第一项。

但是没有理由实际构建列表，所以我们可以使用生成器表达式来代替:

a = next(i for i in userInput if i in wordsTask)

另请注意，如果生成器表达式为空，则会导致异常:StopIteration。您可能想要处理这种情况。或者你可以添加一个默认值

a = next((i for i in userInput if i in wordsTask), 42)