Python itertools.permutations用法及代码示例

用法: itertools.permutations(iterable, r=None)

返回 iterable 中元素的连续 r 长度排列。

如果 r 未指定或为 None ，则 r 默认为 iterable 的长度，并生成所有可能的 full-length 排列。

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

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

大致相当于：

def permutations(iterable, r=None):
    # permutations('ABCD', 2) --> AB AC AD BA BC BD CA CB CD DA DB DC
    # permutations(range(3)) --> 012 021 102 120 201 210
    pool = tuple(iterable)
    n = len(pool)
    r = n if r is None else r
    if r > n:
        return
    indices = list(range(n))
    cycles = list(range(n, n-r, -1))
    yield tuple(pool[i] for i in indices[:r])
    while n:
        for i in reversed(range(r)):
            cycles[i] -= 1
            if cycles[i] == 0:
                indices[i:] = indices[i+1:] + indices[i:i+1]
                cycles[i] = n - i
            else:
                j = cycles[i]
                indices[i], indices[-j] = indices[-j], indices[i]
                yield tuple(pool[i] for i in indices[:r])
                break
        else:
            return

permutations() 的代码也可以表示为 product() 的子序列，过滤以排除具有重复元素的条目(来自输入池中相同位置的条目)：

def permutations(iterable, r=None):
    pool = tuple(iterable)
    n = len(pool)
    r = n if r is None else r
    for indices in product(range(n), repeat=r):
        if len(set(indices)) == r:
            yield tuple(pool[i] for i in indices)

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

相关用法

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