【发布时间】：2017-11-20 10:01:25
【问题描述】：
假设我有四个这样的元素
test = ['A', 'B', 'C', 'D']
然后我可以很容易地得到由三个元素组成的所有组合像这样
import itertools as it
list(it.combinations(test, 3))
给了
[('A', 'B', 'C'), ('A', 'B', 'D'), ('A', 'C', 'D'), ('B', 'C', 'D')]
不过，现在我还有几个后缀，例如
suff = ['x', 'y']
我想考虑一下。我的想要的结果是这样的
[('A__x', 'B__x', 'C__x'),
('A__x', 'B__x', 'C__y'),
('A__x', 'B__x', 'D__x'),
('A__x', 'B__x', 'D__y'),
('A__x', 'B__y', 'C__x'),
('A__x', 'B__y', 'C__y'),
('A__x', 'B__y', 'D__x'),
('A__x', 'B__y', 'D__y'),
('A__x', 'C__x', 'D__x'),
('A__x', 'C__x', 'D__y'),
('A__x', 'C__y', 'D__x'),
('A__x', 'C__y', 'D__y'),
('A__y', 'B__x', 'C__x'),
('A__y', 'B__x', 'C__y'),
('A__y', 'B__x', 'D__x'),
('A__y', 'B__x', 'D__y'),
('A__y', 'B__y', 'C__x'),
('A__y', 'B__y', 'C__y'),
('A__y', 'B__y', 'D__x'),
('A__y', 'B__y', 'D__y'),
('A__y', 'C__x', 'D__x'),
('A__y', 'C__x', 'D__y'),
('A__y', 'C__y', 'D__x'),
('A__y', 'C__y', 'D__y'),
('B__x', 'C__x', 'D__x'),
('B__x', 'C__x', 'D__y'),
('B__x', 'C__y', 'D__x'),
('B__x', 'C__y', 'D__y'),
('B__y', 'C__x', 'D__x'),
('B__y', 'C__x', 'D__y'),
('B__y', 'C__y', 'D__x'),
('B__y', 'C__y', 'D__y')]
所以我想找到两个后缀的所有三个组合，但在这个后缀前面的所有内容中都没有重复（例如，我不想要 ('A__x', 'A__y', 'C__x')，因为会有两个 A）。
我目前实现如下：
import itertools as it
def filter_elements(some_elements, connector):
# get all elements in front of the connector
all_elements = [eli.split(connector)[0] for eli in some_elements]
# return True if all elements are unique
return len(all_elements) == len(set(all_elements))
test = ['A', 'B', 'C', 'D']
suff = ['x', 'y']
connector = '__'
test_suff = ["{}{}{}".format(eli, connector, si) for eli in test for si in suff]
all_combinations = list(it.combinations(test_suff, 3))
desired_combinations = [combi for combi in all_combinations if filter_elements(combi, connector)]
这给了我想要的输出。
显然，这不仅适用于四个元素和两个后缀，还适用于任意组合。有没有更直接的方法可以避免创建all_combinations？
【问题讨论】：
标签： python performance combinations itertools