如何将含多个同级内部循环的嵌套for循环转为map()函数？

先说明：只有当expression1、expression2需要返回结果并收集时，用列表推导或map()才有实际的性能意义；如果只是执行无返回的操作，转换后的写法可能不如原循环直观，性能提升也有限。

列表推导式写法 #

因为两个内层循环是同级关系，我们可以拆成两个独立的推导式，完全对应原代码的逻辑：

1. 收集表达式结果 #

如果需要保存expression1和expression2的执行结果：

# 收集所有符合条件的expression1结果
result1 = [expression1 for item1 in iterable1 for item2 in iterable2 if condition]
# 收集所有符合条件的expression2结果
result2 = [expression2 for item1 in iterable1 for item3 in iterable3 if condition2]

这里的推导式逻辑和原循环完全一致：每个item1都会遍历一遍iterable2/iterable3，只保留满足条件时的表达式结果。

2. 仅执行操作不收集结果 #

如果只是要执行操作不需要保存结果，用生成器表达式（把方括号换成圆括号）再通过list()消耗，比列表推导更节省内存：

# 执行所有符合条件的expression1
list((expression1 for item1 in iterable1 for item2 in iterable2 if condition))
# 执行所有符合条件的expression2
list((expression2 for item1 in iterable1 for item3 in iterable3 if condition2))

map()函数写法 #

map()需要配合可迭代的元素对来处理，这里用itertools.product生成item1与item2/item3的所有组合，再封装逻辑到函数中：

先导入工具库：

import itertools

处理expression1 #

def handle_pair1(pair):
    item1, item2 = pair
    if condition:  # 注意condition要能访问item1、item2的变量，确保作用域正确
        return expression1

# 生成所有item1+item2的组合，用map处理后过滤掉无效结果
result1 = list(filter(None, map(handle_pair1, itertools.product(iterable1, iterable2))))

filter(None)是为了去掉条件不满足时函数返回的None，只保留有效结果。

处理expression2 #

def handle_pair2(pair):
    item1, item3 = pair
    if condition2:
        return expression2

result2 = list(filter(None, map(handle_pair2, itertools.product(iterable1, iterable3))))

额外提示 #

• 列表推导的性能通常比显式for循环好，因为它是Python底层优化过的字节码实现。
• map()的性能和列表推导接近，但如果用自定义普通函数，可能比列表推导略慢；换成lambda表达式的话差异不大。
• 优先优化condition、condition2里的复杂计算逻辑，比单纯换循环结构带来的性能提升更明显。

内容的提问来源于stack exchange，提问作者Apostla