如何使用Python列表推导式拆分test函数返回值得到指定列表？

首先得说清楚为啥你原来的写法不行：[test() for _ in range(5)]会生成一个包含5个(1,2)元组的列表，比如[(1,2), (1,2), (1,2), (1,2), (1,2)]。当你尝试a, b = 这个列表时，Python会试图把列表拆成两个元素，但列表有5个元素，自然会抛出ValueError: too many values to unpack (expected 2)，这就是问题根源。

接下来给你两种可行的方案，按需选择：

方案1：用zip(*...)一次性拆分（推荐，只调用5次test()） #

这种方法只需要调用test()5次，效率更高，适合test()有耗时操作或者副作用的场景：

def test():
    return 1, 2

results = [test() for _ in range(5)]
a, b = zip(*results)
# 如果需要列表类型（zip返回的是元组），可以转成list
a = list(a)
b = list(b)

print(a)  # 输出 [1,1,1,1,1]
print(b)  # 输出 [2,2,2,2,2]

原理是zip(*results)会把列表里的每个元组“解包”并按位置合并——把所有元组的第一个元素凑成一个迭代器，第二个元素凑成另一个，然后我们直接把这两个迭代器分别赋值给a和b。

方案2：分别提取每个位置的元素（调用10次test()） #

如果test()是纯函数（没有副作用、每次返回结果一致），也可以直接在列表推导式里提取元组的对应位置：

def test():
    return 1, 2

a = [test()[0] for _ in range(5)]
b = [test()[1] for _ in range(5)]

print(a)  # 输出 [1,1,1,1,1]
print(b)  # 输出 [2,2,2,2,2]

这种写法更直观，但缺点是会调用test()10次，比方案1多一倍的调用次数，所以如果test()是比较重的操作，不推荐用这个。

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