在列表理解的 else 块中赋值

Question

我想引用两个列表并通过比较生成一个输出列表。

occupied = [8,9,10]
broken = [1,2,3]

output = ['occupied' if x in occupied else x in broken for x in range(1,11)]
desired_output = ['broken', 'broken', 'broken', 'broken', False, False, False, 'occupied', 'occupied', 'occupied']

有没有可能一次达到上面的效果？

目前我正在做两次迭代

['broken' if x2==True else x2 for x2 in ['occupied' if x in occupied else x in broken for x in range(1,11)] ]

我正在寻找这样的东西

['occupied' if x in occupied else 'broken' if x in broken for x in range(1,11)]

但这是不正确的语法

Answer 1

您几乎成功了，只需在您的解决方案中添加一个 else 子句（第二个 if 表达式）：

output = ["broken" if x in broken else 'occupied' if x in occupied else False for x in range(1,11)]
# Out[5]: ['broken', 'broken', 'broken', False, False, False, False, 'occupied', 'occupied', 'occupied']

Answer 2

如果我没理解错的话，这对你应该有用

occupied = [8,9,10]
broken = [1,2,3]
["occupied" if x in occupied else "broken" if x in broken else False for x in range(1,11)]

输出：

['broken',
 'broken',
 'broken',
 False,
 False,
 False,
 False,
 'occupied',
 'occupied',
 'occupied']

在列表理解的 else 块中赋值

assign value in else block of list comprehension

python

list-comprehension

conditional-operator

python-3.x