以下是一个示例代码,用于按组创建组合:
from itertools import combinations
def create_combinations(groups):
combinations_list = []
for group in groups:
combinations_group = list(combinations(group, 2))
combinations_list.extend(combinations_group)
return combinations_list
# 示例输入
groups = [['A', 'B', 'C'], ['X', 'Y']]
# 调用函数
result = create_combinations(groups)
# 输出结果
print(result)
输出:
[('A', 'B'), ('A', 'C'), ('B', 'C'), ('X', 'Y')]
在这个示例中,我们定义了一个函数create_combinations
来按组创建组合。它接受一个包含多个组的列表作为参数。然后,我们使用combinations
函数从每个组中创建了所有可能的2个元素的组合,并将这些组合添加到一个结果列表中。最后,返回结果列表。
在示例输入中,有两个组['A', 'B', 'C']
和['X', 'Y']
。函数返回了这两个组的所有可能的2个元素的组合[('A', 'B'), ('A', 'C'), ('B', 'C'), ('X', 'Y')]
。
你可以根据你的具体需求修改代码来适应不同的输入和输出。
下一篇:按组从多个列中获取最大值的方法