>  기사  >  백엔드 개발  >  Python의 생성기 또는 목록에서 처음 N 항목을 추출하려면 어떻게 해야 합니까?

Python의 생성기 또는 목록에서 처음 N 항목을 추출하려면 어떻게 해야 합니까?

Mary-Kate Olsen
Mary-Kate Olsen원래의
2024-10-20 20:08:30735검색

How Can I Extract the First N Items from a Generator or List in Python?

생성기 또는 목록에서 첫 번째 N 항목을 검색하는 방법은 무엇입니까?

Python에서는 첫 번째 N 요소를 추출하는 다양한 방법이 있습니다. 목록이나 생성기에서. 이 답변은 LINQ의 Take 메서드와 유사점을 참조하여 다양한 접근 방식에 대한 포괄적인 설명을 제공합니다.

1. 목록 분할

LINQ의 Take 메서드와 유사하게 다음 구문을 사용하여 손쉽게 목록을 분할할 수 있습니다.

<code class="python">top5 = array[:5]</code>

이 분할 표기법을 사용하면 시작 및 중지 인덱스를 지정할 수 있습니다. , 선택적 단계 값도 있습니다. 다음과 같은 부분 슬라이스를 얻기 위해 매개변수를 생략할 수 있습니다.

  • array[start:]
  • array[:stop]
  • array[::step]

2. 생성기 슬라이싱

목록과 달리 Python에서는 생성기를 직접 슬라이싱할 수 없습니다. 대신 itertools.islice() 함수를 사용하여 슬라이싱 생성기에서 생성기를 래핑할 수 있습니다. 구문은 다음과 같습니다.

<code class="python">import itertools
top5 = itertools.islice(my_list, 5) # grab the first five elements</code>

생성기를 슬라이싱하면 부분적으로 소진된다는 점을 기억하세요. 전체 생성기를 보존하려면 먼저 튜플이나 목록으로 변환하는 것을 고려하세요.

<code class="python">result = tuple(generator)</code>

위 내용은 Python의 생성기 또는 목록에서 처음 N 항목을 추출하려면 어떻게 해야 합니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

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