Python itertools.combinations用法及代码示例

用法: itertools.combinations(iterable, r)

从输入 iterable 返回元素的 r 长度子序列。

根据输入 iterable 的顺序以字典顺序发出组合元组。因此，如果输入iterable 已排序，则组合元组将按排序顺序生成。

元素根据它们的位置而不是它们的值被视为唯一的。因此，如果输入元素是唯一的，则每个组合中都不会出现重复值。

大致相当于：

def combinations(iterable, r):
    # combinations('ABCD', 2) --> AB AC AD BC BD CD
    # combinations(range(4), 3) --> 012 013 023 123
    pool = tuple(iterable)
    n = len(pool)
    if r > n:
        return
    indices = list(range(r))
    yield tuple(pool[i] for i in indices)
    while True:
        for i in reversed(range(r)):
            if indices[i] != i + n - r:
                break
        else:
            return
        indices[i] += 1
        for j in range(i+1, r):
            indices[j] = indices[j-1] + 1
        yield tuple(pool[i] for i in indices)

combinations() 的代码也可以表示为 permutations() 在过滤元素未按排序顺序(根据它们在输入池中的位置)的条目后的子序列：

def combinations(iterable, r):
    pool = tuple(iterable)
    n = len(pool)
    for indices in permutations(range(n), r):
        if sorted(indices) == list(indices):
            yield tuple(pool[i] for i in indices)

返回的项目数是 n! / r! / (n-r)! 时 0 <= r <= n 或零时 r > n 。

相关用法

注：本文由纯净天空筛选整理自python.org大神的英文原创作品 itertools.combinations。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。