python - 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可以是列表或字符串。

score 3 · Accepted Answer

您需要if集合理解之外的语句，如本else例所示，statements它不是可迭代的

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

score 1 · Accepted Answer

1

试试这个

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

于 2021-05-28T08:28:10.953 回答

score 0 · Accepted Answer

如果您不必不惜一切代价使用集合理解，那么您的代码可能会通过使用.update方法来简化，该方法接受可迭代并且与使用可迭代的每个元素set具有相同的效果。.add简化代码：

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

3 回答 3