Блог

У меня есть список цен на акции компании. Теперь я хочу разделить список на несколько интервалов. мы будем хранить цену следующим образом: первые 2 элемента, затем следующие 3 элемента, затем 2 элемента и так далее.

 meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]
 

Вывод

 meta_stocks = [[10, 9],[11, 15, 19],[22, 25],[ 11, 15, 17]]
 

Я могу разделить список на 5 элементов в каждом, но не могу разделить его дальше

 >>> [meta_stocks[i:i interval2] for i in range(0, len(meta_stocks), interval2)]
>>> [[10, 9, 11, 15, 19], [22, 25, 11, 15, 17]]
 

Вы можете использовать понимание списка с помощью itertools.cycle :

 meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]

from itertools import cycle

start = 0
l = [2,3]
c = cycle(l)

[meta_stocks[start:(start:=start next(c))]
 for i in range(len(l)*len(meta_stocks)//sum(l))]
 

Вывод:

 [[10, 9], [11, 15, 19], [22, 25], [11, 15, 17]]
 

Вы могли бы сделать это так без помощи дополнительного импорта:

 meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]
meta_stocks_out = []
offset = 0
interval = 2
while offset   interval <= len(meta_stocks):
    meta_stocks_out.append(meta_stocks[offset:offset interval])
    offset  = interval
    interval = 2 if interval == 3 else 3
print(meta_stocks_out)
 

Некоторые итерационные инструменты на помощь:

 from itertools import islice, cycle, takewhile

i = iter(meta_stocks)
intervals = [2, 3]

[*takewhile(lambda _: _, ([*islice(i, n)] for n in cycle(intervals)))]
# [[10, 9], [11, 15, 19], [22, 25], [11, 15, 17]]
 

Использование itertools.islice

 from itertools import islice

meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]
chunks = [2, 3, 2, 3                 # length of each sublist

it = iter(meta_stocks)               
result = [list(islice(it, i)) for i in chunks]

print(result)
#Output: [[10, 9], [11, 15, 19], [22, 25], [11, 15, 17]]
 

вы можете попробовать это, сначала мы получаем список, в котором каждый элемент представляет собой список с двумя подсписками из 2 и 3 элементов. После этого мы выравниваем 1-й уровень списка.

 import pprint
meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]
new_list = [item for sublist in [[meta_stocks[i:i 2], meta_stocks[i 2:i 5]] for i in range(0, len(meta_stocks), 5)] for item in sublist]
pprint.pprint(new_list)
 

и вывод

 [[10, 9], [11, 15, 19], [22, 25], [11, 15, 17]]
 

Предположим, что ваши интервалы выражены списком целых чисел

 meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]
intervals = [2,3,2,3]

cintervals = [sum(intervals[:i]) for i in range(0,len(intervals) 1)]

sliced = [meta_stocks[cintervals[i-1]:cintervals[i]] for i in range(1, len(cintervals))]
 

Попробуй:

 meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]

res = [x for i in range(0, len(meta_stocks), 5) for x in [meta_stocks[i:i 2], meta_stocks[i 2:i 5]]]
print(res)
 

Вывод

 [[10, 9], [11, 15, 19], [22, 25], [11, 15, 17]]
 

Общий подход, использующий itertools.accumulate :

 from itertools import accumulate

meta_stocks = [10, 9, 11, 15, 19, 22, 25, 11, 15, 17]

step = sum([2, 3])
steps = list(accumulate([0, 2, 3]))

res = [x for i in range(0, len(meta_stocks), 5) for x in [meta_stocks[i s:i e] for s, e in zip(steps, steps[1:])]]
print(res)