[英]Form all possible combinations of groups of 2, 3, 4 from a list when the length of groups is known
Suppose I have tuple like (1,2,3,4,5)
.假设我有像
(1,2,3,4,5)
这样的元组。
I want to group it like 1,1,0
that is (1 group of 2), (1 group of 3) and (0 groups of 4).我想将它分组为
1,1,0
,即(1 组 2)、(1 组 3)和(0 组 4)。
The result would be (1,2), (3,4,5)
, (2,3), (1,4,5)
, (3,4), (1,2,5)
, (4,5), (1,2,3)
(1,3), (2,4,5)
and so on结果将是
(1,2), (3,4,5)
, (2,3), (1,4,5)
, (3,4), (1,2,5)
, (4,5), (1,2,3)
(1,3), (2,4,5)
等等
How can I implement this?我该如何实施? Is this is possible with
itertools
?这可以用
itertools
吗?
You can use combinations but you will still have to implement your own logic for group partitioning:您可以使用组合,但您仍然必须为组分区实现自己的逻辑:
from itertools import combinations
def group(T,g234):
if sum(g234) == 0: yield [];return # end resursion
size = next(g for g,n in enumerate(g234,2) if n>0) # first count>0
g234 = list(g234) # remaining groups
g234[size-2] -= 1
for combo in combinations(range(len(T)),size): # combine tuple indexes
part = tuple(T[i] for i in combo) # subgroup of combination
others = [v for i,v in enumerate(T) if i not in combo] # remaining
yield from ([part]+rest for rest in group(others,g234)) # assemble parts
output: output:
print(*group((1,2,3,4,5),[1,1,0]),sep="\n")
[(1, 2), (3, 4, 5)]
[(1, 3), (2, 4, 5)]
[(1, 4), (2, 3, 5)]
[(1, 5), (2, 3, 4)]
[(2, 3), (1, 4, 5)]
[(2, 4), (1, 3, 5)]
[(2, 5), (1, 3, 4)]
[(3, 4), (1, 2, 5)]
[(3, 5), (1, 2, 4)]
[(4, 5), (1, 2, 3)]
print(*group((1,2,3,4,5,6),(0,2,0)),sep="\n")
[(1, 2, 3), (4, 5, 6)]
[(1, 2, 4), (3, 5, 6)]
[(1, 2, 5), (3, 4, 6)]
[(1, 2, 6), (3, 4, 5)]
[(1, 3, 4), (2, 5, 6)]
[(1, 3, 5), (2, 4, 6)]
...
print(*group((1,2,3,4,5,6),(3,0,0)),sep="\n")
[(1, 2), (3, 4), (5, 6)]
[(1, 2), (3, 5), (4, 6)]
[(1, 2), (3, 6), (4, 5)]
[(1, 2), (4, 5), (3, 6)]
[(1, 2), (4, 6), (3, 5)]
[(1, 2), (5, 6), (3, 4)]
...
print(*group((1,2,3,4,5,6,7,8,9),(1,1,1)),sep="\n")
[(1, 2), (3, 4, 5), (6, 7, 8, 9)]
[(1, 2), (3, 4, 6), (5, 7, 8, 9)]
[(1, 2), (3, 4, 7), (5, 6, 8, 9)]
[(1, 2), (3, 4, 8), (5, 6, 7, 9)]
...
[EDIT] Iterative version. [编辑]迭代版本。 Uses deque as a replacement for the recursion's call stack:
使用 deque 作为递归调用堆栈的替代品:
from itertools import combinations
from collections import deque
def group(T,g234):
stack = deque([(T,g234,[],None)]) #tuple, groups, parts, combo_iterator
while stack:
T,g234,parts,cIter = stack.pop()
#print(len(T),g234,len(parts),bool(cIter))
if cIter is None:
if not sum(g234): yield parts;continue # partition complete
size = next(g for g,n in enumerate(g234,2) if n>0) # first count>0
g234 = list(g234) # remaining groups
g234[size-2] -= 1
cIter = combinations(range(len(T)),size) # combine tuple indexes
combo = next(cIter,None)
if combo is None: continue
stack.append((T,g234,parts,cIter))
part = tuple(T[i] for i in combo) # subgroup of combination
others = [v for i,v in enumerate(T) if i not in combo] # remaining
stack.append((others,g234,parts+[part],None)) # assemble parts
output: output:
for parts in group(range(1000),(500, 0,0)):
print(parts[:3],"...",parts[-3:])
[(0, 1), (2, 3), (4, 5)] ... [(994, 995), (996, 997), (998, 999)]
[(0, 1), (2, 3), (4, 5)] ... [(994, 995), (996, 998), (997, 999)]
[(0, 1), (2, 3), (4, 5)] ... [(994, 995), (996, 999), (997, 998)]
[(0, 1), (2, 3), (4, 5)] ... [(994, 995), (997, 998), (996, 999)]
[(0, 1), (2, 3), (4, 5)] ... [(994, 995), (997, 999), (996, 998)]
[(0, 1), (2, 3), (4, 5)] ... [(994, 995), (998, 999), (996, 997)]
...
Assuming the original list does not contain duplicate digits then假设原始列表不包含重复的数字,那么
l = {1,2,3,4,5}
[(i,j) for i in itertools.combinations(l,2) for j in itertools.combinations(l.difference(i),3)]
[((1, 2), (3, 4, 5)),
((1, 3), (2, 4, 5)),
((1, 4), (2, 3, 5)),
((1, 5), (2, 3, 4)),
((2, 3), (1, 4, 5)),
((2, 4), (1, 3, 5)),
((2, 5), (1, 3, 4)),
((3, 4), (1, 2, 5)),
((3, 5), (1, 2, 4)),
((4, 5), (1, 2, 3))]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.