python - 在 python 中分发列表

Question

假设我有一个配置字典:

config = {'A': 3, 'B': 4, 'C': 2}

我如何像这样扁平分布(散布)列表:(一个一个追加到结果列表仍然是所有配置的结尾)

result = ['A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'B']

另一个例子:

config = {'A': 3, 'B': 1}
result = ['A', 'B', 'A', 'A']

config = {'A': 2, 'B': 2}
result = ['A', 'B', 'A', 'B']

Answer 1

您可以使用 itertools recipe roundrobin 为此:

from itertools import cycle, islice

def roundrobin(*iterables):
    "roundrobin('ABC', 'D', 'EF') --> A D E B F C"
    # Recipe credited to George Sakkis
    pending = len(iterables)
    nexts = cycle(iter(it).next for it in iterables)
    while pending:
        try:
            for next in nexts:
                yield next()
        except StopIteration:
            pending -= 1
            nexts = cycle(islice(nexts, pending))

result = list(roundrobin(*(k * v for k, v in sorted(config.items()))))