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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。