python ngay bây giờ, tiếp theo, n lần lặp

Question

viết một hàm chung có thể lặp qua bất kỳ vòng lặp nào có thể lặp lại hiện tại, cặp tiếp theo.

def now_nxt(iterable): 
    iterator = iter(iterable) 
    nxt = iterator.__next__() 
    for x in iterator: 
     now = nxt 
     nxt = x 
     yield (now,nxt) 

for i in now_nxt("hello world"): 
    print(i) 

('h', 'e') 
('e', 'l') 
('l', 'l') 
('l', 'o') 
('o', ' ') 
(' ', 'w') 
('w', 'o') 
('o', 'r') 
('r', 'l') 
('l', 'd') 

Tôi đã suy nghĩ về cách tốt nhất để viết một hàm có thể đặt số lượng mục trong mỗi bộ.

ví dụ nếu nó là

func("hello",n=3) 

kết quả sẽ là:

('h','e','l') 
('e','l','l') 
('l','l','o') 

Tôi mới vào sử dụng timeit, vì vậy hãy chỉ ra nếu tôi làm điều gì sai trái ở đây:

import timeit 

def n1(iterable, n=1): 
    #now_nxt_deque 
    from collections import deque 
    deq = deque(maxlen=n) 
    for i in iterable: 
     deq.append(i) 
     if len(deq) == n: 
      yield tuple(deq) 

def n2(sequence, n=2): 
    # now_next 
    from itertools import tee 
    iterators = tee(iter(sequence), n) 
    for i, iterator in enumerate(iterators): 
     for j in range(i): 
      iterator.__next__() 
    return zip(*iterators) 

def n3(gen, n=2): 
    from itertools import tee, islice 
    gens = tee(gen, n) 
    gens = list(gens) 
    for i, gen in enumerate(gens): 
     gens[i] = islice(gens[i], i, None) 
    return zip(*gens) 


def prin(func): 
    for x in func: 
     yield x 

string = "Lorem ipsum tellivizzle for sure ghetto, consectetuer adipiscing elit." 

print("func 1: %f" %timeit.Timer("prin(n1(string, 5))", "from __main__ import n1, string, prin").timeit(100000)) 
print("func 2: %f" %timeit.Timer("prin(n2(string, 5))", "from __main__ import n2, string, prin").timeit(100000)) 
print("func 3: %f" %timeit.Timer("prin(n3(string, 5))", "from __main__ import n3, string, prin").timeit(100000)) 

kết quả:

$ py time_this_function.py 
func 1: 0.163129 
func 2: 2.383288 
func 3: 1.908363 

Answer 1

đề nghị của tôi sẽ là,

from collections import deque 

def now_nxt_deque(iterable, n=1): 
    deq = deque(maxlen=n) 
    for i in iterable: 
     deq.append(i) 
     if len(deq) == n: 
      yield tuple(deq) 

for i in now_nxt_deque("hello world", 3): 
    print(i) 

('h', 'e', 'l') 
('e', 'l', 'l') 
('l', 'l', 'o') 
('l', 'o', ' ') 
('o', ' ', 'w') 
(' ', 'w', 'o') 
('w', 'o', 'r') 
('o', 'r', 'l') 
('r', 'l', 'd') 

Answer 2

Giải pháp của tôi:

def nn(itr, n): 
    iterable = iter(itr) 

    last = tuple(next(iterable, None) for _ in xrange(n)) 
    yield last 
    for _ in xrange(len(itr)): 
     last = tuple(chain(last[1:], [next(iterable)])) 
     yield last 

này đã được thực hiện cho Python 2, nếu bạn muốn sử dụng nó với Python 3, thay thế xrange với range.

next, có default tham số vĩ đại, mà sẽ được trả lại thay vì nuôi một StopIteration, bạn cũng có thể thêm này mặc định tham số để hàm của bạn như sau:

def nn(itr, n, default=None): 
    iterable = iter(itr) 

    last = tuple(next(iterable, default) for _ in xrange(n)) 
    yield last 
    for _ in xrange(len(itr)): 
     last = tuple(chain(last[1:], [next(iterable, default)])) 
     yield last 

tôi đã chơi một số chi tiết với nó, ví dụ sử dụng itr.__class__() làm mặc định, nhưng điều đó có vẻ sai đối với danh sách và bộ dữ liệu, điều đó cũng có ý nghĩa đối với các chuỗi.

Answer 3

Đây là một cách thực sự đơn giản để làm điều đó:

• Clone iterator của bạn n lần sử dụng sử dụng itertools.tee
• Tạm ứng các iterator thứ ii lần
• izip họ tất cả cùng nhau

import itertools 

def now_next(sequence, n=2): 
    iterators = itertools.tee(iter(sequence), n) 
    for i, iterator in enumerate(iterators): 
     for j in range(i): 
      iterator.next() 
    return itertools.izip(*iterators) 

Answer 4

Một biến thể của kỹ thuật của Eric có sử dụng slicing

from itertools import tee, islice, izip 

def now_next(gen, n=2): 
    gens = tee(gen, n) 
    gens = list(gens) 
    for i, gen in enumerate(gens): 
    gens[i] = islice(gens[i], i, None) 
    return izip(*gens) 

for x in now_next((1,2,3,4,5,6,7)): 
    print x 

Answer 5

Một one-liner dựa trên câu trả lời của cravoori:

from itertools import tee, islice, izip 

def now_next(gen, n=2): 
    return izip(*(islice(g, i, None) for i, g in enumerate(tee(gen, n))))