使用一组迭代步骤迭代基于列表的列表

我想根据存储在另一个列表中的可变数量的迭代和存储为整数的恒定数量的跳过来迭代给定的列表。

假设我有三件事 -

1. l- 我需要迭代（或过滤）的列表
2. w- 一个列表，告诉我在休息之前要迭代多少项
3. k- 一个整数，告诉我在每组迭代之间要跳过多少个元素。

换句话说，w 表示要进行多少次迭代，在每组迭代之后，k 表示要跳过多少个元素。

因此，如果 w = [4,3,1] 且 k = 2。那么在给定列表（长度为 14）上，我想迭代前 4 个元素，然后跳过 2 个，然后是下 3 个元素，然后跳过 2 个，然后是下一个 1 个元素，然后跳过 2 个。

另一个例子，

#Lets say this is my original list

l = [6,2,2,5,2,5,1,7,9,4]
w = [2,2,1,1]
k = 1

基于 w 和 k，我想迭代为 -

6 -> Keep # w says keep 2 elements 
2 -> Keep
2 -> Skip # k says skip 1
5 -> Keep # w says keep 2 elements
2 -> Keep
5 -> Skip # k says skip 1
1 -> Keep # w says keep 1 element
7 -> Skip # k says skip 1
9 -> Keep # w says keep 1 element
4 -> Skip # k says skip 1

我尝试从 itertools、numpy、嵌套循环的组合中找到一些东西，但我似乎无法理解如何迭代它。很抱歉没有提供任何尝试，但我不知道从哪里开始。

我不一定需要完整的解决方案，只需一些提示/建议即可。

这有效：

l = [6,2,2,5,2,5,1,7,9,4]
w = [2,2,1,1]
k = 1

def take(xs, runs, skip_size):
    ixs = iter(xs)
    for run_size in runs:
        for _ in range(run_size ):
            yield next(ixs)
        for _ in range(skip_size):
            next(ixs)

result = list(take(l, w, k))
print(result)

Result:

[6, 2, 5, 2, 1, 9]

该函数就是所谓的生成器，一次生成结果的一部分，这就是为什么它被组合成一个列表的原因list(take(l, w, k)).

在函数内部，列表xs传入的内容被包装在迭代器中，以便能够一次获取一项next().

runs定义要获取和产出的物品数量，skip_size定义每次“运行”后要跳过的项目数。

作为奖励，这里有一句有趣的俏皮话 - 如果您能弄清楚它为何有效，我想您已经足够了解这个问题了，可以继续前进了:)

[y for i, y in zip([x for xs in [[1] * aw + [0] * k for aw in w] for x in xs], l) if i]