>백엔드 개발 >파이썬 튜토리얼 >Python 목록에서 겹치는 쌍(또는 삼중)을 효율적으로 반복하려면 어떻게 해야 합니까?

Python 목록에서 겹치는 쌍(또는 삼중)을 효율적으로 반복하려면 어떻게 해야 합니까?

Mary-Kate Olsen
Mary-Kate Olsen원래의
2024-12-16 21:23:11798검색

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

Python에서 겹치는 값 쌍 반복

"현재" 요소와 "다음" 요소를 모두 고려하면서 Python에서 목록을 반복하는 것은 일반적인 필요. 전통적으로 이는 다음과 같은 코드를 사용하여 달성되었습니다.

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

이 접근 방식은 효과적이지만 가장 관용적이거나 효율적이지는 않습니다. 더 파이썬적인 해결책은 itertools 모듈의 pairwise 함수를 사용하는 것입니다:

import itertools

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

pairwise 함수는 두 개의 병렬 반복자 a와 b를 생성하며 둘 다 원래 반복 가능 항목의 첫 번째 요소를 가리킵니다. 그런 다음 b를 한 단계씩 진행하여 a는 s0을 가리키고 b는 s1을 가리킵니다. 그런 다음 zip 함수는 a와 b의 요소를 쌍으로 만들어 겹치는 쌍을 만듭니다.

Python 2의 경우 zip 대신 itertools.izip을 사용하여 지연 반복자를 얻을 수 있습니다.

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)

이 기술은 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)

te를 사용하면 반복자 중 하나가 다른 반복자보다 더 많이 진행되는 경우 메모리가 소모될 수 있습니다. 그러나 반복자가 동일한 속도로 진행되는 경우 이 접근 방식은 메모리 효율적이고 관용적입니다.

위 내용은 Python 목록에서 겹치는 쌍(또는 삼중)을 효율적으로 반복하려면 어떻게 해야 합니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.