python列表推导式嵌套条件推导式使用
时间: 2023-05-26 10:04:04 浏览: 113
Python中的嵌套列表推导式和条件推导式可以结合使用,以产生更复杂的列表。
嵌套列表推导式允许您在一个列表推导式中使用另一个列表推导式。以下是一个简单的例子,该例子将一个矩阵转换为一个平面列表:
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flat = [x for row in matrix for x in row]
print(flat)
输出:[1, 2, 3, 4, 5, 6, 7, 8, 9]
上述示例中的条件推导式是将matrix中的每一行元素同时迭代到一个新的列表row中,并将row中的每个元素x添加到一个新的列表flat中。因此,flat中的每个元素都来自于matrix中的原始矩阵元素。
可以使用类似的技术,将条件表达式嵌套到列表推导式中。这使得仅根据元素类型的条件仅包含特定项成为可能。
以下是一个简单的示例,该示例仅从一个列表中收集所有偶数值:
evens = [x for x in range(10) if x % 2 == 0]
print(evens)
输出:[0, 2, 4, 6, 8]
在这个例子中,for语句产生了0到9的可迭代范围。if语句过滤了仅包含2的整数倍的元素,并将它们追加到evens列表中。
下面的示例展示了如何将嵌套条件推导式添加到矩阵列表推导式中,以仅获取矩阵中的偶数值:
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
evens_only = [x for row in matrix for x in row if x % 2 == 0]
print(evens_only)
输出:[2, 4, 6, 8]
在这个例子中,for语句和if语句嵌套在一起,以产生一个包含仅矩阵中所有偶数值的平面列表。
阅读全文