Iterate over 'zipped' диапазоны массива numpy

Question

Я часто работаю с массивами numpy, представляющими критические времена в временном ряду. Затем я хочу перебирать диапазоны и запускать на них операции. Например:

rngs = [0, 25, 36, 45, ...] 
output = [] 
for left, right in zip(rngs[:-1], rngs[1:]): 
     throughput = do_stuff(array[left:right])... 
     output.append(throughput) 

Есть ли менее неловкий способ сделать это?

Answer 1

Вы можете использовать генератор enumerate

rngs = [0, 25, 36, 45, ...] 
output = [] 
for index, _ in enumerate(rngs[:-1]): 
     throughput = do_stuff(array[index:index+1])... 
     output.append(throughput) 

в одной строке со списком понимания:

rngs = [0, 25, 36, 45, ...] 
output = [do_stuff(array[index:index+1]) for index, _ in enumerate(rngs[:-1])]