Как через каждые N элементов удалить N элементов из списка?

Question

[haniaman]

Допустим, есть список, где 10000 чисел, и надо сделать, чтобы через каждую тысячу чисел было удалено 300 последующих чисел. Как это можно реализовать?

Comments

[Тимур Покровский]

то есть нужно удалить 1000-1300, 2000-2300, 3000-3300...?

[haniaman]

Тимур Покровский, смысл легче будет объяснить на меньших числах. Например всего 20 чисел, и через каждые 3 числа надо удалить 2 числа, например возьмем список count = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20], и на выходе мы получим список [1,2,3,6,7,8,11,12,13,16,17,18]. И тоже самое, только в больших размерах

Answer 1

[0xD34F]

N = 1000
M = 300
NM = N + M

newArr = sum((arr[i * NM:(i + 1) * NM - M] for i in range(ceil(len(arr) / NM))), [])

Comments

[haniaman]

спасибо, именно то что мне и надо было

Answer 2

[aRegius]

import math
import itertools

data_for_test = list(range(1, 31))
ALL_MASK_VALS = 10
TRUE_MASK_VALS = 3
MASK_LEN = math.ceil(len(data_for_test)/ALL_MASK_VALS)
 
mask = (1 if i < TRUE_MASK_VALS else 0 for _ in range(MASK_LEN) for i in range(ALL_MASK_VALS))
result = list(itertools.compress(data_for_test, mask))

>>> result
[1, 2, 3, 11, 12, 13, 21, 22, 23]