Python 列表理解 elif

在 Python 中，我们可以使用列表理解（List Comprehension）非常方便地生成列表。其中包含了 if 和 for 的语法结构，可以用来过滤和变换列表中的元素。

但是有时候，我们需要在列表理解中使用 elif 来进一步筛选元素，这时候该怎么办呢？本文将简单介绍一下如何在列表理解中使用 elif。

基本语法

在列表理解中使用 elif 的语法结构如下：

[expression1 if condition1 else expression2 if condition2 else expression3 for item in iterable]

其中，item 是 iterable 中的元素，expression1 和 expression2 是根据 condition1 和 condition2 判断是否执行的表达式，expression3 是默认情况下（即不满足前面的条件）的表达式。

举个例子，假设我们有一个列表 lst，里面包含一些数字，我们需要把其中大于等于 5 的元素设为 1，小于 5 的元素设为 0，那么可以使用以下语法：

lst = [1, 2, 3, 4, 5, 6, 7, 8, 9]

new_lst = [1 if x >= 5 else 0 for x in lst]

print(new_lst)
# 输出：[0, 0, 0, 0, 1, 1, 1, 1, 1]

以上代码中，我们使用了 if/else 条件判断表达式，并把满足条件的数字替换成了 1，不满足条件的数字替换成了 0。

使用 elif

如果我们需要在列表理解中使用 elif，那么需要把条件判断语句拆分成多个 if/elif/else 子句，从而实现更复杂的逻辑。

举个例子，假设我们有一个字符串列表 words，我们需要把其中长度大于 5 的字符串变成大写字母，长度在 3-5 之间的字符串变成小写字母，长度小于 3 的字符串不变，那么可以使用以下语法：

words = ['apple', 'orange', 'banana', 'peach', 'grape']

new_words = [word.upper() if len(word) > 5 else word.lower() if len(word) >=3 else word for word in words]

print(new_words)
# 输出：['apple', 'orange', 'banana', 'peach', 'GRAPE']

以上代码中，我们使用了 if/elif/else 条件判断表达式，并根据字符串长度来进行大小写转换。需要注意的是，多个 if/elif/else 子句之间的优先级应该自己确定，并且对于 elif 子句必须使用 else 来标记最后一个表达式。

总结

以上就是在 Python 列表理解中使用 elif 的简单介绍，通过条件判断表达式可以实现复杂的逻辑，但需要注意好各个子句之间的优先级和标记位置。对于更复杂的情况，可以使用 if 表达式来编写更清晰和简洁的代码。