두 개의 배열을 동시에 반복하는 pythonic 방법은 무엇입니까? 나는 같은 길이의 두 개의 배열을 가지고있는 경우

Question

-a 및 b이

a = [4,6,2,6,7,3,6,7,2,5]

일반적으로

b = [6,4,6,3,2,7,8,5,3,5]

, 내가 지금처럼 이런 짓을 했을까 말 :

for i in range(len(a)): 
    print a[i] + b[i] 

보다는 뭔가를 이렇게 :

i=0 
for number in a: 
    print number + b[i] 
    i += 1 

나는 사용 된 방법과 일관되기를 선호하기 때문에.

나는 zip을 알고 있지만 결코 사용하지 않습니다. 이것은 zip을 위해 만들어진 것입니까?

는

for pair in zip(a,b): 
    print pair[0] + pair[1] 

이 작업을 수행 할 수있는 파이썬 방법이 될 것이다?

Answer 1

(@Vincenzo PII가 켰을 때) a 및 b이 짧은 목록은 zip를 사용하는 경우 :

for x, y in zip(a, b): 
    print(x + y) 

을 목록 a 및 b가 긴 경우, 메모리 저장 itertools.izip를 사용

import itertools as IT 
for x, y in IT.izip(a, b): 
    print(x + y) 

을

zip은 튜플 목록을 생성합니다. a 및 b이 큰 경우 부담이 될 수 있습니다 (메모리 단위).

itertools.izip 반복자를 반환합니다. 반복자는 튜플의 전체 목록을 생성하지 않습니다. for-loop에 의해 요청 될 때만 각 항목을 산출합니다. 따라서 메모리를 절약 할 수 있습니다.

짧은 목록의 zip(a,b)을 호출하는 것이 itertools.izip(a,b)을 사용하는 것보다 빠릅니다. 그러나 Python3에서 zip은 기본적으로 반복기를 반환합니다 (즉, Python2에서는 itertools.izip과 같습니다).관심의

---

다른 변종 :

• from future_builtin import zip - 당신이 Python3 스타일 Python2에 zip 동안 함께 프로그램하고자하는 경우.
• itertools.izip_longest - a 및 b의 길이가 동일하지 않은 경우.

Answer 2

자신을 언급 한 바와 같이 가능한 솔루션은 zip을 사용하지만 약간 다르게 당신이 질문을 작성하는 방법에 비해 :

는

for x, y in zip(a, b): 
    print x, y 

공지 사항 zip()에 의해 반환 된 튜플의리스트의 길이가 동일 할 것 a과 b의 길이 사이의 최소값까지. a과 b의 길이가 같지 않을 때 영향을줍니다.

Answer 3

zip 대신에 Numpy을 사용할 수 있습니다. 특히 속도가 중요하고 긴 배열이있는 경우에는 특히 유용합니다. 그것 훨씬 더 빨리 당신은 당신이 루프를하지 않아도, 그냥 쓸 수 NumPy와 배열을 사용하면 :

print a + b 

을 표시하는 그래프는 우편, izip 및 NumPy와를 사용하여 길이가 다른 목록을 합산위한 타이밍 평균 : 를

Answer 4

numpy 이래로이 답변을 완벽하게 제공하는 것은 다른 답변에서 논의되었으며, 더 높은 순위의 배열에서 값을 함께 연결하는 것이 유용한 경우가 많습니다.

accepted answer 시퀀스가 ​​listlist의 S 같이 또한 이러한 2 레벨 이상의 numpy 어레이와 같은 다중 레벨 (들이지만, 경우하지만 랭크 1의 모든 시퀀스/배열 움직인다 또는), 각 순위를 반복해야합니다.

import numpy as np 
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) 
b = np.array([list('abc'), list('pdq'), list('xyz')]) 
c = np.array([[frobnicate(aval, bval) for aval, bval in zip(arow, brow)] for arow, brow in zip(a, b)]) 

와 동일한 형상의 이차원 중첩 시퀀스 중 어느 세트에 대해 작동 동일한 개념 : 아래 차원 numpy 배열 예이다

a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 
b = [list('abc'), list('pdq'), list('xyz')] 
c = [[frobnicate(aval, bval) for aval, bval in zip(arow, brow)] for arow, brow in zip(a, b)] 

하나 중첩의 양이면

from itertools import zip_longest as zipl 
a = [[], [4, 5, 6], [7, 8, 9]] # empty list in the first row 
b = [list('abc'), list('pdq'), []] # empty list in the last row 
c = [[frobnicate(aval, bval) for aval, bval in zipl(arow, brow)] for arow, brow in zipl(a, b)]