如何获取范围内的列表输出

Question

如何在范围内获得列表输出？

假设n=4和k=3

那么我们如何得到列表[1，2，1，1]，其中数组的长度为n，k为范围(1….k(

ex-2

如果n＝5并且k＝3

列表将为[1,2,3,1,2]

Answer 1

您可以使用itertools.cycle对从1到k的值进行循环，然后使用iterttools.islice从中获取n个值：

from itertools import cycle, islice
k = 3
n = 7
k_cycle = cycle(range(1, k+1))
out = list(islice(k_cycle, n))
print(out)
# [1, 2, 3, 1, 2, 3, 1]

Answer 2

短期解决方案：

[i % k + 1 for i in range(n)]

以及另一个稍长的解决方案：

[i for j in range((n + k - 1) // k) for i in range(1, k + 1)][:n]

与上述相似，但表达不同

(list(range(1, k + 1)) * ((n + k - 1) // k))[:n]

Answer 3

试试这个

n,k = map(int,input().strip().split())
x=[]
for i in range(n):
x.append(1+i%k)
print(x)

Answer 4

试试这个：

n = 5
k = 3
current_number = 1
ls = []
for count in range(n):
ls.append(current_number)
current_number = ((current_number) % k) + 1

Answer 5

具有itertools.cycle的解决方案

def rangelist(n, k):
return [i for i,_ in zip(cycle(range(1,k+1)), range(n))]
rangelist(n=5,k=3)
>>> [1, 2, 3, 1, 2]

Answer 6

您可以使用itertools:完成

import itertools  
k,n = 3, 5    
result = []    
# defining iterator  
iterators = itertools.cycle(list(range(1,k+1)))  
# for in loop  
for i in range(n):    
x.append(next(iterators))

打印(x(