能否通过Python列表推导式一次迭代生成两个目标列表？

当然可以实现！你想要的一次迭代、仅调用一次函数就拆分出两个目标列表的需求，Python里有非常简洁高效的实现方式——用生成器表达式配合zip(*...)来完成。

具体实现代码 #

def my_func(i):
    return i * 2, 'a' * i

# 生成两个列表，仅一次迭代、一次函数调用
first_vals, second_vals = map(list, zip(*(my_func(i) for i in range(10))))

为什么这个方法可行？ #

这里的核心逻辑是：

1. (my_func(i) for i in range(10))是一个生成器表达式，它会逐个迭代range(10)的元素，调用my_func并返回元组，不会提前把所有元组存入内存（相比方案3的中间列表更省内存）。
2. zip(*...)相当于对生成器输出的元组进行“转置”操作：把每个元组的第一个元素集中到一组，第二个元素集中到另一组。
3. 最后用map(list, ...)把zip返回的迭代器转换成真正的列表（如果不需要列表，只需要可迭代对象的话，可以省略这一步）。

对比你提到的三种方案 #

• 方案1：两次调用my_func，如果函数是耗时操作（比如涉及IO、复杂计算），这种写法的效率会非常低，完全不推荐。
• 方案2：循环写法直观易懂，适合新手，但代码相对冗长，不如生成器+zip的写法简洁。
• 方案3：仅调用一次函数，但会生成一个中间元组列表，当数据量很大时会占用额外内存，不如生成器版本高效。

额外小提示 #

如果你的my_func返回的元组元素更多（比如3个或以上），这个方法同样适用，只需要对应增加接收变量即可：

val1, val2, val3 = map(list, zip(*(my_func(i) for i in range(10))))

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