ทำซ้ำรายการเป็นคู่ (ปัจจุบันถัดไป) ใน Python

บางครั้งฉันต้องวนซ้ำรายการใน Python โดยดูที่องค์ประกอบ "ปัจจุบัน" และองค์ประกอบ "ถัดไป" จนถึงตอนนี้ฉันทำด้วยรหัสเช่น:

for current, next in zip(the_list, the_list[1:]):
    # Do something

สิ่งนี้ได้ผลและทำในสิ่งที่ฉันคาดหวัง แต่มีวิธีที่เป็นสำนวนหรือมีประสิทธิภาพมากกว่าในการทำสิ่งเดียวกันนี้หรือไม่?

นี่คือตัวอย่างที่เกี่ยวข้องจากเอกสารโมดูลitertools :

import itertools
def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = itertools.tee(iterable)
    next(b, None)
    return zip(a, b)   

สำหรับ Python 2 คุณต้องitertools.izipใช้แทนzip:

import itertools
def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = itertools.tee(iterable)
    next(b, None)
    return itertools.izip(a, b)

วิธีการทำงาน:

ขั้นแรกให้ทำซ้ำสองตัวแบบขนานaและbถูกสร้างขึ้น (การtee()เรียก) ซึ่งทั้งสองชี้ไปที่องค์ประกอบแรกของการทำซ้ำดั้งเดิม ตัววนซ้ำตัวที่สองbถูกย้ายไปข้างหน้า 1 ก้าว (การnext(b, None)) เรียก) ณ จุดนี้ชี้aไปที่ s0 และbชี้ไปที่ s1 ทั้งสองaและbสามารถสำรวจตัววนซ้ำดั้งเดิมได้อย่างอิสระ - ฟังก์ชัน izip รับตัววนซ้ำสองตัวและสร้างคู่ขององค์ประกอบที่ส่งคืนโดยเลื่อนตัววนซ้ำทั้งสองในจังหวะเดียวกัน

ข้อแม้อย่างหนึ่ง: tee()ฟังก์ชันนี้สร้างตัวทำซ้ำสองตัวที่สามารถเลื่อนไปข้างหน้าอย่างอิสระจากกัน แต่มีค่าใช้จ่าย หากตัวทำซ้ำตัวใดตัวหนึ่งก้าวหน้ากว่าอีกตัวหนึ่งแสดงว่าtee() จำเป็นต้องเก็บองค์ประกอบที่ใช้ไปไว้ในหน่วยความจำจนกว่าตัววนซ้ำตัวที่สองจะเข้าใช้งานด้วย (ไม่สามารถ 'ย้อนกลับ' ตัววนซ้ำเดิมได้) ที่นี่ไม่สำคัญเพราะตัววนซ้ำหนึ่งก้าวล้ำหน้าอีก 1 ก้าว แต่โดยทั่วไปแล้วมันง่ายที่จะใช้หน่วยความจำจำนวนมากด้วยวิธีนี้

และเนื่องจากtee()สามารถใช้nพารามิเตอร์ได้จึงสามารถใช้กับตัวทำซ้ำแบบขนานได้มากกว่าสองตัว:

def threes(iterator):
    "s -> (s0,s1,s2), (s1,s2,s3), (s2, s3,4), ..."
    a, b, c = itertools.tee(iterator, 3)
    next(b, None)
    next(c, None)
    next(c, None)
    return zip(a, b, c)

ม้วนเอง!

def pairwise(iterable):
    it = iter(iterable)
    a = next(it, None)

    for b in it:
        yield (a, b)
        a = b

เนื่องจากthe_list[1:]สร้างสำเนาของรายการทั้งหมด (ไม่รวมองค์ประกอบแรก) และzip()สร้างรายการสิ่งที่เพิ่มขึ้นทันทีเมื่อมีการเรียกระบบจะสร้างสำเนารายการของคุณทั้งหมดสามชุด หากรายการของคุณมีขนาดใหญ่มากคุณอาจต้องการ

from itertools import izip, islice
for current_item, next_item in izip(the_list, islice(the_list, 1, None)):
    print(current_item, next_item)

ซึ่งไม่ได้คัดลอกรายการเลย

ฉันแค่พูดออกไปฉันแปลกใจมากที่ไม่มีใครคิดจะแจกแจง ()

for (index, thing) in enumerate(the_list):
    if index < len(the_list):
        current, next_ = thing, the_list[index + 1]
        #do something

การวนซ้ำโดยดัชนีสามารถทำสิ่งเดียวกัน:

#!/usr/bin/python
the_list = [1, 2, 3, 4]
for i in xrange(len(the_list) - 1):
    current_item, next_item = the_list[i], the_list[i + 1]
    print(current_item, next_item)

เอาท์พุท:

(1, 2)
(2, 3)
(3, 4)

ตอนนี้เป็นการนำเข้าอย่างง่าย ณ วันที่ 16 พฤษภาคม 2020

from more_itertools import pairwise
for current, next in pairwise(your_iterable):
  print(f'Current = {current}, next = {nxt}')

เอกสารสำหรับเครื่องมือเพิ่มเติม ภายใต้ประทุนรหัสนี้เหมือนกับในคำตอบอื่น ๆ แต่ฉันชอบการนำเข้ามากเมื่อมี

หากคุณยังไม่ได้ติดตั้ง: pip install more-itertools

ตัวอย่าง

ตัวอย่างเช่นหากคุณมีลำดับ fibbonnacci คุณสามารถคำนวณอัตราส่วนของคู่ที่ตามมาได้ดังนี้

from more_itertools import pairwise
fib= [1,1,2,3,5,8,13]
for current, nxt in pairwise(fib):
    ratio=current/nxt
    print(f'Curent = {current}, next = {nxt}, ratio = {ratio} ')

จับคู่จากรายการโดยใช้ความเข้าใจในรายการ

the_list = [1, 2, 3, 4]
pairs = [[the_list[i], the_list[i + 1]] for i in range(len(the_list) - 1)]
for [current_item, next_item] in pairs:
    print(current_item, next_item)

เอาท์พุท:

(1, 2)
(2, 3)
(3, 4)

ฉันประหลาดใจมากที่ไม่มีใครพูดถึงวิธีแก้ปัญหาทั่วไปที่สั้นกว่าง่ายกว่าและที่สำคัญที่สุด:

Python 3:

from itertools import islice

def n_wise(iterable, n):
    return zip(*(islice(iterable, i, None) for i in range(n)))

Python 2:

from itertools import izip, islice

def n_wise(iterable, n):
    return izip(*(islice(iterable, i, None) for i in xrange(n)))

ทำงานสำหรับการวนซ้ำแบบคู่โดยการส่งผ่านn=2แต่สามารถจัดการกับจำนวนที่สูงกว่า:

>>> for a, b in n_wise('Hello!', 2):
>>>     print(a, b)
H e
e l
l l
l o
o !

>>> for a, b, c, d in n_wise('Hello World!', 4):
>>>     print(a, b, c, d)
H e l l
e l l o
l l o
l o   W
o   W o
  W o r
W o r l
o r l d
r l d !

วิธีแก้ไขเบื้องต้น:

def neighbors( list ):
  i = 0
  while i + 1 < len( list ):
    yield ( list[ i ], list[ i + 1 ] )
    i += 1

for ( x, y ) in neighbors( list ):
  print( x, y )

code = '0016364ee0942aa7cc04a8189ef3'
# Getting the current and next item
print  [code[idx]+code[idx+1] for idx in range(len(code)-1)]
# Getting the pair
print  [code[idx*2]+code[idx*2+1] for idx in range(len(code)/2)]