python - 根据列表创建给定字符串的所有可能组合

Question

你们会如何根据给定的字符串创建组合？

示例:我有一位名叫约翰的同事，我想看看他与其他 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 数据库中。我尝试更改我的 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)