Python 列表推导中的 if/else

在本文中，我们将介绍如何在 Python 的列表推导式中使用 if/else 条件语句。列表推导式是一种简洁且强大的语法，在创建新的列表时非常有用。if/else 语句允许我们在列表推导式中根据条件选择要包含的元素，从而实现更灵活的列表生成。

阅读更多：Python 教程

列表推导式简介

列表推导式是一种简化创建列表的语法，它允许我们使用一行代码生成一个新的列表。它的基本语法形式是：

new_list = [expression for item in old_list]

其中，expression 是对 item 执行的操作，old_list 是原始列表。通过这样的语法形式，我们可以在一个简洁的表达式中对原列表的每个元素进行处理，并生成新的列表。

在列表推导式中使用 if 语句

if 语句允许我们在列表推导式中添加一个条件，只有当条件满足时才将元素添加到新的列表中。例如，我们可以使用列表推导式生成一个包含原列表偶数元素的新列表:

old_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
new_list = [x for x in old_list if x % 2 == 0]
print(new_list)

输出结果为:

[2, 4, 6, 8, 10]

在上面的示例中，我们使用 if x % 2 == 0 条件语句过滤了原列表中的奇数，并将满足条件的偶数添加到新列表中。

在列表推导式中使用 if/else 语句

if/else 语句允许我们在列表推导式中根据条件选择要包含的元素。基本的语法形式如下:

new_list = [expression1 if condition else expression2 for item in old_list]

其中 expression1 是条件满足时要执行的操作，expression2 是条件不满足时要执行的操作，condition 是条件表达式。通过这种形式，我们可以根据一个条件选择要在新列表中添加的不同元素。

下面是一个示例，生成一个包含原列表各个元素的平方或立方的新列表:

old_list = [1, 2, 3, 4, 5]
new_list = [x**2 if x % 2 == 0 else x**3 for x in old_list]
print(new_list)

输出结果为:

[1, 4, 27, 16, 125]

在上面的示例中，我们使用了 if x % 2 == 0 else x**3 条件语句，在满足条件时将元素的平方添加到新列表，否则将元素的立方添加到新列表。

列表推导式的嵌套使用

除了在列表推导式中使用 if/else 语句外，我们还可以嵌套多个列表推导式来生成更复杂的列表。

以下示例生成了一个包含二维数组中所有元素的新列表:

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
new_list = [x for row in matrix for x in row]
print(new_list)

输出结果为:

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

在上面的示例中，我们嵌套了两个 for 循环，用于遍历二维数组的每个元素，并将它们添加到新的一维列表中。

总结

在本文中，我们介绍了如何在 Python 的列表推导式中使用 if/else 语句。if/else 语句允许我们根据条件选择要包含的元素，使得列表推导式更加灵活和强大。通过学习列表推导式的相关知识，我们可以以更简洁的方式创建列表，提高代码的可读性和效率。

我们了解了基本的列表推导式语法，并学会使用 if 语句在列表推导式中过滤元素。我们还学习了如何使用 if/else 语句在列表推导式中根据条件选择要包含的元素，并演示了列表推导式的嵌套使用生成更复杂的列表。

希望本文对你理解和应用 Python 中的列表推导式有所帮助，并增加了对 if/else 条件语句在列表推导式中的灵活运用的认识。通过熟练掌握列表推导式的使用，可以提高我们的编程效率，并使得代码更加简洁和易读。