给定列表中的字符串，创建所有可能的组合

Question

你们将如何创建给定字符串的组合？

示例：我有一个名为john的同事，我想看看他和其他3个同事一起时的样子。

公司人数为136人。

Teammates = ['john', 'lizzy', 'tom', 'sarah', 'tiffany', 'max', 'james', 'alice', 'bob']

预期产量：

Which co worker?: John

['john', 'lizzy', 'tom', 'sarah',]
['john', 'lizzy', 'tom', 'max']
['john', 'lizzy', 'tom', 'james']
['john', 'lizzy', 'tom', 'alice']
['john', 'lizzy', 'tom', 'bob']
['john', 'tiffany', 'max', 'james']

我试过使用itertools排列。 我拥有的名称存储在sql db中。 我尝试更改sql查询以对目标合作者进行优先级排序，并针对其他合作者进行迭代。

Answer 1

您可以为此使用itertools.combinations：

import itertools as it

def teammate_combinations(who, group_size)
    return (x for x in it.combinations(teammates, group_size) if who in x)

teammate_combinations('john', 4)