python - 在python中为列表中的成员赋值

Question

我有一个列表，让我们说:

test = [False, False, 2, False, False, False, 3, False, False]

并且我想将此列表的每个成员分配给 False 除非该成员等于 2所以结果应该是:

test = [False, False, 2, False, False, False, False, False, False]

我试过了:

test = [False for i in test if i !=2]

但结果是:

[False, False, False, False, False, False, False, False]

然后我尝试定义一个函数:

def set_False(lst):
     for i in lst:
         if i != 2:
            i = False
    return lst

当我申请时:

test = set_False(test)

结果是:

[False, False, 2, False, False, False, 3, False, False]

我不能在我放置代码的环境中使用枚举，所以有什么建议吗？

谢谢

Answer 1

是

[False if i !=2 else 2 for i in test]

否则，当它等于 2 时，您将跳过整个元素。