如何重复列出直到一定数量的项目

Question

我正在用python索引数据集，并希望生成一个列表，该列表重复某些数字直到列表中的某些项目为止

我的数据集有很多行，我想生成一个[1,2,3,4,5]的列表，重复进行直到达到行数为止（在这种情况下为90，但是我d希望使代码不明确）。 因此，直到此列表中有90个术语，它就像[1,2,3,4,5,1,2,3,4,5,1,2,3,4,5,1，...] 。 最有效的方法是什么？我尝试过*

Answer 1

使用cycle，它可以迭代并无限重复：

from itertools import cycle

r = cycle([1, 2, 3, 4, 5])
result = [next(r) for _ in range(90)]
print(result)

输出：

[1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, ...]

Answer 2

您可以尝试以下方法：

lst = [i % 5 + 1 for i in range(N)]

N是您想要的项数（例如90）