在 Python 列表理解中使用带有 if/elif 语句的“for”循环 [重复]

Question

我正在尝试将此 for 循环转换为列表理解：

a = [1,2,3,4,5,6,7,8,9]
result = []
for i in a:
    if i <= 3:
        result.append(1)
    elif i > 4 and i < 7:
        result.append(2)

我已经试过了

[1 if i <= 3 else 2 if i > 3 and i < 7 for i in a]

抱怨

File "<ipython-input-155-eebf07a9e0d8>", line 2
    [1 if i <= 3 else 2 if i > 3 and i < 7 for i in a]
                                             ^
SyntaxError: invalid syntax

Answer 1

列表理解：

添加更多条件：D（不，这真的很混乱）

[
    1 if i <= 3 else 2
    for i in a
    if i != 4 and i < 7
]

我们是怎么到这里的？

基本列表组合：[EXPRESSION for TARGET in ITERABLE if CONDITION]

三元表达式：(IF_TRUE if CONDITION else IF_FALSE)

1. 获取 for 循环。很简单 for i in a。
2. 添加过滤掉将被忽略的项目的条件。一旦超过CONDITION，列表中的该位置必须有一个项目。在这种情况下，我们不希望 i 为 4 或大于 7。if i != 4 and i < 7。
3. 对项目做你需要的事情。在这种情况下，如果i 小于或等于4，我们需要1。否则，我们将取2。1 if i <= 3 else 2。注意：这是一个三元表达式。检查出来！