python - 从迭代中生成列表列表

Question

这是我要解决的问题的示例：

d=[1,2,3]
e=[d[0],d[1]]

def add_up(x,y):
    return (x + y)

我的尝试：

h=[[add_up(a,b)] for a in e for b in d]
print h

我尝试的输出

[[2], [3], [4], [3], [4], [5]]

我想要的输出：**[[2,3,4],[3,4,5]]**

有什么建议么？谢谢。

Answer 1

你快到了，使用：

print [[add_up(a, b) for b in d] for a in e]  # prints [[2, 3, 4], [3, 4, 5]]