파이썬 반복자에 대한 인덱스

Question

나는 반복자 iterator과 인덱스 목록 indices (가능한 반복)을 가지고 있으며, 나는 그 반복자에서 그 요소들을 추출하고 싶다. 지금 내가하고있는 순간에

indices = sorted(indices) 
deltas = [indices[0]] + [indices[i+1] - indices[i] for i in range(len(indices) - 1)] 
output = [] 
for delta in deltas: 
    for i in range(delta): 
     datum = next(iterator) 
    output.append(datum) 

루프의 두 레이어가 필요합니까? itertools으로 트릭을 놓치고 있습니까?

Answer 1

당신은 확실히 더블 루프가 필요하지 않습니다 당신은 C에 의해 수표의 비용을 줄일 수

[datum for i, datum in enumerate(x) if i in indices] 

: 당신이합니다 (sort을하지만 피) 검사에 대한 오버 헤드 발생으로 지수의 매우 낮은 번호에 대한 지능형리스트 전환 indices ~ set.

indices = set(indices) 
[datum for i, datum in enumerate(x) if i in indices] 

업데이트 : 첫 번째와 세 번째 옵션은 겨우 900 밀리 초에서 타이밍과 거의 비슷하다 (약간 나는 set 건설을 통해 sort의 성능을보고 관심이있을 것이다 (세트 조회는 O (1)이다) 처음부터 끝까지) 10,000,000 개의 항목 중에서 1000 개의 임의의 색인을 선택합니다. OP의 코드는 약 1.2 초 만에 실행되었습니다.

Answer 2

메모리가 제약 조건이 아니라면 max 인덱스를 찾고 해당 최대 인덱스까지 반복자 값의 배열을 채울 것입니다. 어쨌든 중간 값을 계산해야하므로 델타를 계산하여 실제로 아무것도 얻지 못합니다.

이

it = iter(sorted(indices)) 
index = next(it) 
for i, datum in enumerate(iterator): 
    if i != index: 
     continue 
    output.append(datum) 
    try: 
     index = next(it) 
    except StopIteration: 
     break 

또한이 작업을 수행 할 수 있습니다 단일 루프 및 생성 델타없이이 작업을 수행 할 수 있지만 체크 코드를 더 복잡하게로

max_index = max(indices) 
data = [v for v in itertools.islice(iterator, max_index + 1)] 
values = [data[i] for i in indices]