Python所有数据组合

时间:2013-08-25 18:41:05

标签: python list combinations

所以我有一个类似的文件结构(每行数据)

A
B
C
...

我想阅读这个文件的每一行并将它们组合成所有可能的组合,我正在寻找的结果就是这个(我会放在一个列表中)

A
AB
AC
ABC
ACB
B
BA
BC
BCA
BAC
C
CA
CB
CBA
CAB

显然,我想要的不仅仅是那三条线......

感谢

1 个答案:

答案 0 :(得分:5)

您应该使用itertools库。您想要生成powerset中每个元素的所有唯一排列。有些代码可能看起来像

from itertools import permutations, combinations, chain

# Taken from itertools page, but edited slightly to not return empty set
def powerset(iterable):
    "powerset([1,2,3]) --> (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(iterable)
    return chain.from_iterable(combinations(s, r) for r in range(1, len(s)+1))

然后

In [1]: s = ('A', 'B', 'C')

In [2]: [j for i in powerset(s) for j in permutations(i)]
Out[2]: 
[('A',),
 ('B',),
 ('C',),
 ('A', 'B'),
 ('B', 'A'),
 ('A', 'C'),
 ('C', 'A'),
 ('B', 'C'),
 ('C', 'B'),
 ('A', 'B', 'C'),
 ('A', 'C', 'B'),
 ('B', 'A', 'C'),
 ('B', 'C', 'A'),
 ('C', 'A', 'B'),
 ('C', 'B', 'A')]