python - 用步骤穿插列表

Question

我有一个帖子列表`[Post1, Post2, Post3, Post4, ..., PostN] 我需要在其中穿插广告(基本上，在 N 个帖子之后插入)。例如:

step = 3
insert_advertisement(posts_list, advertisement, step) # Insert advertisement after 3 posts

输出:

[Post1, Post2, Post3, 广告, Post4, Post5, Post6, 广告, ...]

有没有用 itertools 实现这个的好方法？如果没有 - 那么我将不胜感激任何其他解决方案。

Answer 1

使用生成器函数:

>>> def insert_advertisement(posts_list, advertisement, step):
...     for i, post in enumerate(posts_list):
...         if i > 0 and i % step == 0:
...             yield advertisement
...         yield post
...
>>> list(insert_advertisement([1, 2, 3, 4, 5, 6, 7], 'ad', 3))
[1, 2, 3, 'ad', 4, 5, 6, 'ad', 7]