迹忆客 专注技术分享

当前位置:主页 > 学无止境 > 编程语言 >

在 Python 的列表推导中使用 elif 语句

作者:迹忆客 最近更新:2023/01/25 浏览次数:

要在列表推导中使用 elif 语句:

  1. 使用 if 语句检查条件。
  2. 使用 else 语句来实现 elif 子句。
  3. 如果两个条件都不满足,则使用第二个 else 语句返回一个值。
a_list = [1, 2, 2, 5, 1, 9]

new_list = [
    'a' if item == 1
    else 'b' if item == 2 else 'c'
    for item in a_list
]
print(new_list)  # 👉️ ['a', 'b', 'b', 'c', 'a', 'c']

列表推导用于对每个元素执行某些操作或选择满足条件的元素子集。

我们使用 else 语句在列表理解中实现 elif 子句。

if 语句检查当前项是否等于 1,如果满足条件,则返回 aelse 语句返回另一个条件。 条件检查项目是否等于 2。如果满足条件,则返回 b

如果两个条件都不满足,则返回 c。

在伪代码中,语法如下所示。

new_list = [
  return A if condition_is_met
  else return B if condition_is_met else return C
  for item in a_list
]

请注意ifelse 语句位于列表理解的开头。

以下是我们如何使用 for 循环实现示例中的列表理解。

a_list = [1, 2, 2, 5, 1, 9]

new_list = []

for item in a_list:
    if item == 1:
        new_list.append('a')
    elif item == 2:
        new_list.append('b')
    else:
        new_list.append('c')

# 👇️ ['a', 'b', 'b', 'c', 'a', 'c']
print(new_list)

代码示例中的 for 循环等同于列表推导。

这是在列表推导中使用 elif 语句的另一个示例。

a_list = [1, 2, 2, None, 1, None]

new_list = [
    0 if item is None
    else item if item % 2 == 0 else item + 11
    for item in a_list
]
print(new_list)  # 👉️ [12, 2, 2, 0, 12, 0]

if 语句检查当前项是否为 None,如果满足条件,则返回 0。else 语句实现另一个条件。 该条件检查数字除以 2 的余数是否为 0。

如果满足条件,则返回该项目,否则,返回该 item + 11

如果我们只需要在列表理解中实现 if-else 语句,则可以使用相同的方法。

a_list = [1, 2, 2, None, 1, None]

new_list = [
    0 if item is None
    else item
    for item in a_list
]
print(new_list)  # 👉️ [1, 2, 2, 0, 1, 0]

if 语句检查项目是否为 None。 如果满足条件,则返回 0,否则,返回项目。

伪代码中的语法如下所示。

new_list = [
  return A if condition_is_met
  else return B
  for item in list
]

请注意 ,if 和 else 语句是在列表理解的开头指定的。

如果列表理解中只有 if 语句,请确保在末尾指定它。

a_list = [1, 2, 2, None, 1, None]

new_list = [
    item for item in a_list
    if item is not None
]
print(new_list)  # 👉️ [1, 2, 2, 1]

在列表推导中实现 if-elif-else 语句可能会使您的代码更难阅读。

如果是这种情况,只需使用 for 循环即可。

a_list = [1, 2, 2, None, 1, None]


new_list = []

for item in a_list:
    if item is None:
        new_list.append(0)
    elif item % 2 == 0:
        new_list.append(item)
    else:
        new_list.append(item + 11)

print(new_list)  # 👉️ [12, 2, 2, 0, 12, 0]

for 循环方法有点冗长,但在我看来,它更具可读性和直观性。

本站文章均为原创或翻译,转载请发邮件至 1244347461@qq.com 进行申请,未经许可,禁止转载。经作者同意之后,转载请以链接形式注明出处

本文地址:

扫一扫阅读全部技术教程

社交账号
  • https://www.github.com/onmpw
  • qq:1244347461

近期修改

教程更新

热门标签

扫码一下
查看教程更方便