Python-Как создать список из N элементов из данного списка

Question

Мне нужно создать список размеров N из уже предоставленного списка.

N = 3 
a_list = [10,4,18,2,6,19,24,1,20] 

O/P должно быть:

[10,4,18] [4,18,2] [18,2,6] [2,6,19] [6,19,24] [19,24,1] [24,1,20] 

Это как размер окна N = 3, который скользит один шаг справа.

Как мне это сделать?

Answer 1

Более быстрый способ:

>>> zip(a_list,a_list[1:],a_list[2:]) 
[(10, 4, 18), (4, 18, 2), (18, 2, 6), (2, 6, 19), (6, 19, 24), (19, 24, 1), (24, 1, 20)] 

Сравнение:

In [6]: %timeit [a_list[i:i+n] for i in xrange(len(a_list)-n+1)] 
100000 loops, best of 3: 9.61 us per loop 

In [7]: %timeit zip(a_list,a_list[1:],a_list[2:]) 
100000 loops, best of 3: 5.23 us per loop 

Или более общий:

>>> zip(*[a_list[i:] for i in range(3)]) #3 (or 2, 4, 5, etc)is the length of step 

Для больших list, вероятно, вам придется использовать numpy, чтобы получить более быстрое решение, чем @Ashwini Чаудхари (в http://www.rigtorp.se/2011/01/01/rolling-statistics-numpy.html):

import numpy as np 
lista=np.array(lis) 
def rolling_window(a, window): 
    shape = a.shape[:-1] + (a.shape[-1] - window + 1, window) 
    strides = a.strides + (a.strides[-1],) 
    return np.lib.stride_tricks.as_strided(a, shape=shape, strides=strides) 

%timeit [lis[i:i+n] for i in xrange(len(lis)-n+1)] 
%timeit rolling_window(lista, n) 

1 loops, best of 3: 171 ms per loop 
100000 loops, best of 3: 19.7 µs per loop 

Answer 2

Используйте список понимание и нарезка:

>>> lis = [10,4,18,2,6,19,24,1,20] 
>>> n = 3 
>>> [lis[i:i+n] for i in xrange(len(lis)-n+1)] 
[[10, 4, 18], [4, 18, 2], [18, 2, 6], [2, 6, 19], [6, 19, 24], [19, 24, 1], [24, 1, 20]] 
>>> n = 4 
>>> [lis[i:i+n] for i in xrange(len(lis)-n+1)] 
[[10, 4, 18, 2], [4, 18, 2, 6], [18, 2, 6, 19], [2, 6, 19, 24], [6, 19, 24, 1], [19, 24, 1, 20]] 

Для большего списка подход, основанный на zip на самом деле медленнее:

In [27]: n = 100                     

In [28]: lis = [10,4,18,2,6,19,24,1,20]*10000              

In [30]: %timeit zip(*[lis[i:] for i in xrange(n)])            
1 loops, best of 3: 593 ms per loop                

In [31]: %timeit [lis[i:i+n] for i in xrange(len(lis)-n+1)]          
10 loops, best of 3: 114 ms per loop  

Answer 3

попробовать этот

>>> lis = [10,4,18,2,6,19,24,1,20] 
>>> n=3 
>>> [lis[i:i+n] for i in range(len(lis))][:-2] 

выход

[[10, 4, 18], [4, 18, 2], [18, 2, 6], [2, 6, 19], [6, 19, 24], [19, 24, 1], [24, 1, 20]]