时间戳

首页 > 解决方案 > 具有多个级别的 islice 和循环

python - 具有多个级别的 islice 和循环

问题描述

更新：按要求添加了所需的模式我有 2 个列表，预期输出与上次不同

Numberset1 = [10,11,12]
Numberset2 = [1,2,3,4,5]

我想通过操作列表来显示输出，预期的输出是

10 1 1
10 1 2
10 1 3
10 1 4
10 1 5
10 2 2
10 2 3
10 2 4
10 2 5
10 2 1
10 3 3
10 3 4
10 3 5
10 3 1
10 3 2
10 4 4
10 4 5
10 4 1
10 4 2
10 4 3
10 5 5
10 5 1
10 5 2
10 5 3
10 5 4
11 2 2
11 2 3
11 2 4
11 2 5
11 2 1
11 3 3
11 3 4
11 3 5
11 3 1
11 3 2
11 4 4
11 4 5
11 4 1
11 4 2
11 4 3
11 5 5
11 5 1
11 5 2
11 5 3
11 5 4
11 5 1
11 1 1
11 1 2
11 1 3
11 1 4
11 1 5
12 3 3
12 3 4
12 3 5
12 3 1
12 3 2
12 4 4
12 4 5
12 4 1
12 4 2
12 4 3
12 4 4
12 4 5
12 5 5
12 5 1
12 5 2
12 5 3
12 1 1
12 1 2
12 1 3
12 1 4
12 1 5
12 2 2
12 2 3
12 2 4
12 2 5
12 2 1

我尝试过的代码如下，这是在上一个问题中提出的，我尝试将它用于下一级循环，但我无法获得所需的输出

Numberset1 = [10,11,12]
Numberset2 = [1,2,3,4,5]

from itertools import cycle, islice

it = cycle(Numberset2)
for i in Numberset1:
    for a in Numberset2:
        for j in islice(it, len(Numberset2)):
            print(i, a,j)
        skipped1 = next(it)
    skipped1 = next(it)

我得到的输出是

10 1 1
10 1 2
10 1 3
10 1 4
10 1 5
10 2 2
10 2 3
10 2 4
10 2 5
10 2 1
10 3 3
10 3 4
10 3 5
10 3 1
10 3 2
10 4 4
10 4 5
10 4 1
10 4 2
10 4 3
10 5 5
10 5 1
10 5 2
10 5 3
10 5 4
11 1 2
11 1 3
11 1 4
11 1 5
11 1 1
11 2 3
11 2 4
11 2 5
11 2 1
11 2 2
11 3 4
11 3 5
11 3 1
11 3 2
11 3 3
11 4 5
11 4 1
11 4 2
11 4 3
11 4 4
11 5 1
11 5 2
11 5 3
11 5 4
11 5 5
12 1 3
12 1 4
12 1 5
12 1 1
12 1 2
12 2 4
12 2 5
12 2 1
12 2 2
12 2 3
12 3 5
12 3 1
12 3 2
12 3 3
12 3 4
12 4 1
12 4 2
12 4 3
12 4 4
12 4 5
12 5 2
12 5 3
12 5 4
12 5 5
12 5 1

请注意当数字 11 在第一列开始时的变化而不是预期的输出

我们如何将循环和islice用于多个级别

模式：第一列应按 Numberset1 中的数字顺序，Numberset1 中第一个数字的第二列应按 Numberset2 中的数字顺序，Numberset1 中第一个数字的第三列应按 NUMberset2 中的数字顺序，但当Numberset1 中第一个数字的第二列发生更改，它也应该从 Numberset2 列表中的第二个数字更改和打印，依此类推

标签： pythonpython-3.xitertools

解决方案

这是一个使用cycleand完成任务的版本islice。为了使代码更简洁，我创建了一个生成器函数aligned_cycle，它循环遍历产生的项目，cycle直到我们得到我们想要开始当前循环的项目。

此更新版本可以处理Numberset1比Numberset2.

from itertools import cycle, islice

def aligned_cycle(seq, start_item):
    ''' Make a generator that cycles over the items in `seq`.
        The first item yielded equals `start_item`.
    '''
    if start_item not in seq:
        raise ValueError("{} not in {}".format(start_item, seq))
    it = cycle(seq)
    for u in it:
        if u == start_item:
            break
    yield u
    yield from it

Numberset1 = [10, 11, 12]
Numberset2 = [1, 2, 3, 4, 5]
cycle_length = len(Numberset2)

for i, u in zip(Numberset1, cycle(Numberset2)):
    for j in islice(aligned_cycle(Numberset2, u), cycle_length):
        for k in islice(aligned_cycle(Numberset2, j), cycle_length):
            print(i, j, k)

输出

10 1 1
10 1 2
10 1 3
10 1 4
10 1 5
10 2 2
10 2 3
10 2 4
10 2 5
10 2 1
10 3 3
10 3 4
10 3 5
10 3 1
10 3 2
10 4 4
10 4 5
10 4 1
10 4 2
10 4 3
10 5 5
10 5 1
10 5 2
10 5 3
10 5 4
11 2 2
11 2 3
11 2 4
11 2 5
11 2 1
11 3 3
11 3 4
11 3 5
11 3 1
11 3 2
11 4 4
11 4 5
11 4 1
11 4 2
11 4 3
11 5 5
11 5 1
11 5 2
11 5 3
11 5 4
11 1 1
11 1 2
11 1 3
11 1 4
11 1 5
12 3 3
12 3 4
12 3 5
12 3 1
12 3 2
12 4 4
12 4 5
12 4 1
12 4 2
12 4 3
12 5 5
12 5 1
12 5 2
12 5 3
12 5 4
12 1 1
12 1 2
12 1 3
12 1 4
12 1 5
12 2 2
12 2 3
12 2 4
12 2 5
12 2 1

Jon Clements编写了一个更健壮、更高效的版本aligned_cycle：

def aligned_cycle(iterable, start_item):
    a, b = tee(iterable)
    b = cycle(b)
    for u, v in zip(a, b):
        if u == start_item:
            break
    else:
        return

    yield u
    yield from b

谢谢，乔恩！

推荐阅读