合并两个集合并考虑每个集合的顺序（在 Python 中）

Question

我有两个变量 i，j，它们显示两个集合 1 和 2 的长度，例如 len(one)=i 和 len(two)=j。现在，我想合并这两个集合，以使每个集合具有有序排列。我还需要在 Python 中为每个新集合建立索引。

例如： 一个包含前 i 个大写字母，两个包含小写字母

 len(one) = i  
    len(two) = j 
expected outputs = {'abABC...', 'aAbBC...', 'aABcC...', 'aABCb...',...}

我尝试了以下代码，但它不起作用。如果有人能提供帮助，我将不胜感激。

    from functools import reduce
    from itertools import combinations

    def assign(v, p):
        v[p[0]] = p[1]
        return v

    def interp(word1, word2, size):
        return (''.join(reduce(assign, zip(comb1, word1), zip(comb2, word2)))
                for comb1, comb2 in zip(combinations(range(size), len(word1)),
                                        combinations(range(size), len(word2))))

    print('\n'.join(interp("ABC", "ab", 5)))

Answer 1

您可以使用一个函数将两个列表之一的第一项与列表其余部分的组合递归地合并：

def merge(a, b):
    if a and b:
        for (first, *rest), other in (a, b), (b, a):
            yield from ([first, *merged] for merged in merge(rest, other))
    elif a or b:
        yield a or b

这样：

for combination in merge(['A', 'B', 'C'], ['a', 'b']):
    print(''.join(combination))

输出：

ABCab
ABabC
ABaCb
AabBC
AaBCb
AaBbC
abABC
aABCb
aABbC
aAbBC

请注意，Python 中的集合是无序的，因此如果您的输入是集合，则无法按照预期输出的建议保留 ABC 和 ab 的顺序。此处给出的示例假定您的输入和输出是列表。

Answer 2

从itertools借用partition配方：

one = set(['A', 'B', 'C'])
two = set(['a', 'b'])

from itertools import permutations, tee, filterfalse, chain

def partition(pred, iterable):
    'Use a predicate to partition entries into false entries and true entries'
    # partition(is_odd, range(10)) --> 0 2 4 6 8   and  1 3 5 7 9
    t1, t2 = tee(iterable)
    return filterfalse(pred, t1), filter(pred, t2)

iter_1 = ((i, 1) for i in one)
iter_2 = ((i, 2) for i in two)

for c in permutations(chain(iter_1, iter_2), 5):
    p1, p2 = map(list, partition(lambda k: k[1] == 1, c))
    if sorted(p1, key=lambda k: k[0]) == p1 and sorted(p2, key=lambda k: k[0]) == p2:
        print(''.join(i[0] for i in c))

打印：

ABCab
ABaCb
ABabC
AaBCb
AaBbC
AabBC
aABCb
aABbC
aAbBC
abABC