Rumah >pembangunan bahagian belakang >Tutorial Python >Bagaimanakah Saya Boleh Mengulangi Pasangan Bertindih (atau Tiga Kali ganda) dengan Cekap dalam Senarai Python?

Bagaimanakah Saya Boleh Mengulangi Pasangan Bertindih (atau Tiga Kali ganda) dengan Cekap dalam Senarai Python?

Mary-Kate Olsen
Mary-Kate Olsenasal
2024-12-16 21:23:11797semak imbas

How Can I Efficiently Iterate Over Overlapping Pairs (or Triples) in a Python List?

Menyalurkan Pasangan Nilai Bertindih dalam Python

Membalas senarai dalam Python sambil mempertimbangkan kedua-dua elemen "semasa" dan "seterusnya" ialah keperluan bersama. Secara tradisinya, ini telah dicapai menggunakan kod seperti:

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

Walaupun berkesan, pendekatan ini bukanlah yang paling idiomatik atau cekap. Penyelesaian yang lebih Pythonic ialah menggunakan fungsi berpasangan daripada modul 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)

Fungsi berpasangan mencipta dua iterator selari, a dan b, kedua-duanya menunjuk kepada elemen pertama bagi lelaran asal. Ia kemudiannya memajukan b dengan satu langkah, menghasilkan satu menunjuk ke s0 dan b menunjuk ke s1. Fungsi zip kemudian memasangkan elemen daripada a dan b, memberikan anda pasangan bertindih.

Untuk Python 2, anda boleh menggunakan itertools.izip dan bukannya zip untuk mendapatkan lelaran malas:

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)

Teknik ini boleh diperluaskan untuk menghasilkan "tetingkap" yang lebih besar menggunakan parameter n fungsi tee. Contohnya, untuk menghasilkan tiga kali ganda, anda boleh menggunakan:

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)

Perhatikan bahawa menggunakan tee boleh menggunakan memori jika salah satu iterator maju lebih jauh daripada yang lain. Walau bagaimanapun, dalam kes di mana iterator maju pada kadar yang sama, pendekatan ini adalah cekap ingatan dan idiomatik.

Atas ialah kandungan terperinci Bagaimanakah Saya Boleh Mengulangi Pasangan Bertindih (atau Tiga Kali ganda) dengan Cekap dalam Senarai Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn