python 中的 If-for-else 嵌套集合理解

Question

我正在尝试将以下嵌套条件转换为设置理解，但无法使其正常工作。

processed = set()
if isinstance(statements, list):
    for action in statements:
        processed.add(action)
else:
    processed.add(statements)

我尝试了以下方法，但看起来我犯了一个错误

processed = {action for action in statements if isinstance(statements, list) else statements}

编辑：其中 statements 可以是列表或字符串。

Answer 1

你需要集合理解之外的 if 语句，在 else 的情况下，statements 不是可迭代的

processed = {action for action in statements} if isinstance(statements, list) else {statements}

Answer 2

试试这个

proceed={x for x in statements} if isinstance(statements, list) else statements

Answer 3

如果您不必不惜任何代价使用集合理解，那么您的代码可能会通过使用 set 的 .update 方法来简化，它接受可迭代的并且与使用 [=] 具有相同的效果13=] 对于可迭代的每个元素。简化代码：

processed = set()
if isinstance(statements, list):
    processed.update(statements)
else:
    processed.add(statements)

If-for-else nested set comprehension in python