函数不存储所有输出

Question

我有以下清单，

p=[list(['a', 'b', 'c']), list(['d', 'e'])]

我想制作每个元素的子集（大小为 2）并列出它们，这将给出如下输出：

[[('a', 'b'), ('a', 'c'), ('b', 'c')],[('d', 'e')]]

为了实现这一点，我编写了以下函数，

def x(m,n):
    for i in x:
        z=list(itertools.combinations(i, n))
    return(z)

然而，当我申请 z(m,2) 时，我只得到最后一个元素：

[('d', 'e')]

我想知道我做错了什么？

Answer 1

这是因为您每次都设置 z 而不是附加它：

def x(m,n):
    z = []
    for i in m:
        z.append(list(itertools.combinations(i, n)))
    return(z)

产量：

[[('a', 'b'), ('a', 'c'), ('b', 'c')], [('d', 'e')]]