从列表创建子列表

时间:2015-04-02 21:59:38

标签: python list

我正在尝试在python中创建一个包含子列表的列表;比如,一组的适当子集。例如,

A = [1, 2, 3, 4]

Desired List = [[1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4], [1, 2, 3], [1, 2, 4], [2, 3, 4]]

谢谢!

2 个答案:

答案 0 :(得分:1)

看起来你想从列表中获得所有组合。尝试使用itertools.combinations

desired_list = itertools.combinations(A, 2)

答案 1 :(得分:1)

因为您似乎只想要大小为2或更大的子集:

from itertools import combinations, chain
A = range(1, 5)
list(chain(*(combinations(A, r) for r in range(2, len(A)))))
# [(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4), (1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]

如果您想要所有正确的子集,只需将range(2, len(A))更改为range(len(A))