我正在用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,...] 。 最有效的方法是什么?我尝试过*
答案 0 :(得分:3)
使用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, ...]
答案 1 :(得分:0)
您可以尝试以下方法:
lst = [i % 5 + 1 for i in range(N)]
N
是您想要的项数(例如90)