[英]Python periodically remove items from a list
假设我的清单很长,包含1000个元素,并且我想根据两个变量定期删除元素组。
因此,对于my_list=[1,2,3,4...1000]
和a=5
, b=7
,我将保留前5个元素,删除下7个元素,然后重复执行直到列表末尾。
列表将如下所示:
my_list = [1,2,3,4,5,12,13,14,15,16...]
在使用它们之前,我不知道a或b,也不知道列表的长度,因此我正在寻找一种通用的解决方案。
谢谢!
这是使用enumerate
并以a
和b
的和作为索引的mod
的一种方法。 筛选出mod小于a
:
l = range(1, 30)
a, b = 5, 7
r = [x for i, x in enumerate(l) if i%(a+b) < a]
print(r)
# [1, 2, 3, 4, 5, 13, 14, 15, 16, 17, 25, 26, 27, 28, 29]
PS,如果您要删除下一个7,则不应包含12。
这是使用迭代器的另一种方法:
import itertools as it
L = list(range(1000))
a, b = 5, 7
mask = it.chain(it.repeat(1, a), it.repeat(0, b))
Result = list(it.compress(L, it.cycle(mask)))
这里很酷的事情是, 没有中间数据存储在内存中,并且所有结果都是即时生成的。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.