islice和循环有多个级别

时间:2018-05-04 18:43:13

标签: python python-3.x itertools

更新: 添加了所要求的模式 我有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中第一个数字的第3列应按NUmberset2中的数字顺序排列,但是第二列时对于Numberset1更改中的第一个数字,它也应该从Numberset2列表中的2ndnumber更改和打印,依此类推

1 个答案:

答案 0 :(得分:3)

这是使用cycleislice完成任务的版本。为了使代码更清晰,我创建了一个生成器函数aligned_cycle,它循环遍历由cycle产生的项,直到我们得到我们想要开始当前循环的那个。

此更新版本可以处理长度超过Numberset1的{​​{1}}。

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)

Jon Clements撰写了更强大,更高效的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 版本:

aligned_cycle

谢谢,乔恩!