我有一个返回两个值的函数，我想使用列表理解来填充两个列表.
def f(x):
return 2*x,x*x
x = range(3)
xlist, ylist = [f(value) for value in x]
EDITS from answers below:
xtuple, ytuple = zip(*[f(value) for value in x])
xlist, ylist = map(list,zip(*[f(value) for value in x]))
预期回报率应为:
xlist = [0, 2, 4]
ylist = [0, 1, 4]
我的问题归结为:
目前我得到了一个元组列表，虽然这是合理的，但我最终需要两个独立的列表.目前，我可以有1个占位符(元组列表)变量和3个总理解.但我想知道是否有一个干净的方法来做这件事，作为单一列表理解.
值得注意的是:在实际代码中，我的两个返回是相关的，所以我不能简单地将函数拆分为两个.
首先，你犯了一个小错误:应该是:
[f(value) for value in x]
# ^ notice the `value`
而不是:
[f(x) for value in x]
此外，重点是:
return 2*x,x
缩写为:
return (2*x,x)
所以a tuple.你的列表不是列表的元组.带星号的zip和is that you can easily use it in reverse是件好事:
xlist,ylist = zip(*[f(value) for value in x])
# ^ with asterisk注意，xlist和ylist将是tuples(因为zip将被解包).如果希望它们成为列表，可以使用:
xlist,ylist = map(list,zip(*[f(value) for value in x]))其结果是:
>>> xlist
[0, 2, 4]
>>> ylist
[0, 1, 4]
(请注意，range从0开始计数)
Alternative:另一种方法当然是:
xlist = [f(value)[0] for value in x]
ylist = [f(value)[1] for value in x]
但这当然是inelegantly，而且效率可能很低(给定f在计算上很昂贵).